ResNet18Small¶

class cl_gym.backbones.ResNet18Small(multi_head=True, num_classes_per_head=5, num_classes=100)[source]¶

Bases: cl_gym.backbones.resnet.ResNet

ResNet-18 with 1/3 less feature maps, a common backbone in CL literature. See Appendix C in https://openreview.net/pdf?id=Fmg_fQYUejf for further details.

blocks: Union[Iterable[nn.Module], nn.ModuleList]¶

multi_head: bool¶

num_classes_per_head: int¶

training: bool¶

CNN1D Trainer