from gan_training.models import (
    resnet4,
    resnet4_AdaFM_bias,
    resnet4_AdaFM_bias_classCondition,
    resnet4_AdaFM_accumulate_multitasks,
    resnet4_AdaFM_accumulate_lowrank_highLayers, 
)

generator_dict = {
    'resnet4': resnet4.Generator,
    'resnet4_AdaFM_bias': resnet4_AdaFM_bias.Generator,
    'resnet4_AdaFM_bias_classCondition': resnet4_AdaFM_bias_classCondition.Generator,
    'resnet4_AdaFM_accumulate_multitasks': resnet4_AdaFM_accumulate_multitasks.Generator,
    'resnet4_AdaFM_accumulate_lowrank_highLayers': resnet4_AdaFM_accumulate_lowrank_highLayers.Generator,
}

discriminator_dict = {
    'resnet4': resnet4.Discriminator,
    'resnet4_AdaFM_bias': resnet4_AdaFM_bias.Discriminator,
    'resnet4_AdaFM_bias_classCondition': resnet4_AdaFM_bias_classCondition.Discriminator,
    'resnet4_AdaFM_accumulate_multitasks': resnet4_AdaFM_accumulate_multitasks.Discriminator,
    'resnet4_AdaFM_accumulate_lowrank_highLayers': resnet4_AdaFM_accumulate_lowrank_highLayers.Discriminator,
}

encoder_dict = {
}

342bf1aa8147307c73e71904954a4d19e90903ee
<mask token>

<mask token>
generator_dict = {'resnet4': resnet4.Generator, 'resnet4_AdaFM_bias':
    resnet4_AdaFM_bias.Generator, 'resnet4_AdaFM_bias_classCondition':
    resnet4_AdaFM_bias_classCondition.Generator,
    'resnet4_AdaFM_accumulate_multitasks':
    resnet4_AdaFM_accumulate_multitasks.Generator,
    'resnet4_AdaFM_accumulate_lowrank_highLayers':
    resnet4_AdaFM_accumulate_lowrank_highLayers.Generator}
discriminator_dict = {'resnet4': resnet4.Discriminator,
    'resnet4_AdaFM_bias': resnet4_AdaFM_bias.Discriminator,
    'resnet4_AdaFM_bias_classCondition': resnet4_AdaFM_bias_classCondition.
    Discriminator, 'resnet4_AdaFM_accumulate_multitasks':
    resnet4_AdaFM_accumulate_multitasks.Discriminator,
    'resnet4_AdaFM_accumulate_lowrank_highLayers':
    resnet4_AdaFM_accumulate_lowrank_highLayers.Discriminator}
encoder_dict = {}

from gan_training.models import resnet4, resnet4_AdaFM_bias, resnet4_AdaFM_bias_classCondition, resnet4_AdaFM_accumulate_multitasks, resnet4_AdaFM_accumulate_lowrank_highLayers
generator_dict = {'resnet4': resnet4.Generator, 'resnet4_AdaFM_bias':
    resnet4_AdaFM_bias.Generator, 'resnet4_AdaFM_bias_classCondition':
    resnet4_AdaFM_bias_classCondition.Generator,
    'resnet4_AdaFM_accumulate_multitasks':
    resnet4_AdaFM_accumulate_multitasks.Generator,
    'resnet4_AdaFM_accumulate_lowrank_highLayers':
    resnet4_AdaFM_accumulate_lowrank_highLayers.Generator}
discriminator_dict = {'resnet4': resnet4.Discriminator,
    'resnet4_AdaFM_bias': resnet4_AdaFM_bias.Discriminator,
    'resnet4_AdaFM_bias_classCondition': resnet4_AdaFM_bias_classCondition.
    Discriminator, 'resnet4_AdaFM_accumulate_multitasks':
    resnet4_AdaFM_accumulate_multitasks.Discriminator,
    'resnet4_AdaFM_accumulate_lowrank_highLayers':
    resnet4_AdaFM_accumulate_lowrank_highLayers.Discriminator}
encoder_dict = {}

from gan_training.models import (
    resnet4,
    resnet4_AdaFM_bias,
    resnet4_AdaFM_bias_classCondition,
    resnet4_AdaFM_accumulate_multitasks,
    resnet4_AdaFM_accumulate_lowrank_highLayers, 
)

generator_dict = {
    'resnet4': resnet4.Generator,
    'resnet4_AdaFM_bias': resnet4_AdaFM_bias.Generator,
    'resnet4_AdaFM_bias_classCondition': resnet4_AdaFM_bias_classCondition.Generator,
    'resnet4_AdaFM_accumulate_multitasks': resnet4_AdaFM_accumulate_multitasks.Generator,
    'resnet4_AdaFM_accumulate_lowrank_highLayers': resnet4_AdaFM_accumulate_lowrank_highLayers.Generator,
}

discriminator_dict = {
    'resnet4': resnet4.Discriminator,
    'resnet4_AdaFM_bias': resnet4_AdaFM_bias.Discriminator,
    'resnet4_AdaFM_bias_classCondition': resnet4_AdaFM_bias_classCondition.Discriminator,
    'resnet4_AdaFM_accumulate_multitasks': resnet4_AdaFM_accumulate_multitasks.Discriminator,
    'resnet4_AdaFM_accumulate_lowrank_highLayers': resnet4_AdaFM_accumulate_lowrank_highLayers.Discriminator,
}

encoder_dict = {
}
