<filename>models/__init__.py
#from .vgg import *
from .masked_vgg import *
from .masked_psg_vgg import *
from .seed_vgg import *
from .psg_seed_vgg import *

from .dpn import *
from .lenet import *
from .senet import *
from .pnasnet import *
from .densenet import *
from .googlenet import *
from .seed_resnet import *
from .seed_wide_resnet import *
from .shufflenet import *
from .shufflenetv2 import *
#from .resnet import *
from .masked_resnet import *
from .masked_wide_resnet import *
from .masked_psg_wide_resnet import *
from .psg_seed_wide_resnet import *
from .resnext import *
from .preact_resnet import *
from .mobilenet import *
from .mobilenetv2 import *
from .efficientnet import *

# PSG
# from .psg_resnet import *
from .masked_psg_resnet import *
from .psg_seed_resnet import *

from .masked_imagenet_resnet import *
from .masked_psg_imagenet_resnet import *

from .masked_convs import *
from .masked_psg_convs import *


<filename>models/__init__.py
#from .vgg import *
from .masked_vgg import *
from .masked_psg_vgg import *
from .seed_vgg import *
from .psg_seed_vgg import *

from .dpn import *
from .lenet import *
from .senet import *
from .pnasnet import *
from .densenet import *
from .googlenet import *
from .seed_resnet import *
from .seed_wide_resnet import *
from .shufflenet import *
from .shufflenetv2 import *
#from .resnet import *
from .masked_resnet import *
from .masked_wide_resnet import *
from .masked_psg_wide_resnet import *
from .psg_seed_wide_resnet import *
from .resnext import *
from .preact_resnet import *
from .mobilenet import *
from .mobilenetv2 import *
from .efficientnet import *

# PSG
# from .psg_resnet import *
from .masked_psg_resnet import *
from .psg_seed_resnet import *

from .masked_imagenet_resnet import *
from .masked_psg_imagenet_resnet import *

from .masked_convs import *
from .masked_psg_convs import *


#from .vgg import * #from .resnet import * # PSG # from .psg_resnet import *