FS-TFP/federatedscope/cv/model/__init__.py

9 lines
290 B
Python

from __future__ import absolute_import
from __future__ import print_function
from __future__ import division
from federatedscope.cv.model.cnn import ConvNet2, ConvNet5, VGG11
from federatedscope.cv.model.model_builder import get_cnn
__all__ = ['ConvNet2', 'ConvNet5', 'VGG11', 'get_cnn']