MCSGD¶

class cl_gym.algorithms.MCSGD(backbone, benchmark, params)[source]¶

Bases: cl_gym.algorithms.base.ContinualAlgorithm

Mode Connectivity SGD

By Mirzadeh et al. :https://openreview.net/forum?id=Fmg_fQYUejf

calculate_line_loss(w_start, w_end, loader)[source]¶

calculate_point_loss(net, loader)[source]¶

find_connected_minima(task)[source]¶

training_epoch_end()[source]¶

training_step(task_id, inp, targ, optimizer, criterion)[source]¶

training_task_end()[source]¶

OGD Multitask