__init__.py 724 B

12345678910111213141516171819
  1. """
  2. The :py:mod:`models` module provides a set of (eventually pre-trained) models
  3. that can be used for your continual learning experiments and applications.
  4. These models are mostly `torchvision.models
  5. <https://pytorch.org/vision/0.8/models.html#torchvision-models>`_ and `pytorchcv
  6. <https://pypi.org/project/pytorchcv/>`_ but we plan to add more architectures in
  7. the near future.
  8. """
  9. from .simple_cnn import *
  10. from .simple_mlp import *
  11. from .pytorchcv_wrapper import *
  12. from .mlp_tiny_imagenet import SimpleMLP_TinyImageNet
  13. from .mobilenetv1 import MobilenetV1
  14. from .dynamic_modules import *
  15. from .utils import *
  16. from .slda_resnet import SLDAResNetModel
  17. from .icarl_resnet import *
  18. from .ncm_classifier import NCMClassifier