def getModel(self, modelType, dataset="CIFAR100"):
if modelType=="densenet":
if dataset=="MNIST":
print ("MNIST dataset not supported in this model. Try resnet20 or 32")
assert(False)
return dn.DenseNet(growthRate=12, depth=40, reduction=0.5,
bottleneck=True, nClasses=100)
elif modelType=="resnet32":
if dataset=="MNIST":
return res.resnet32mnist(10)
return res.resnet32(100)
elif modelType=="resnet20":
if dataset=="MNIST":
return res.resnet20mnist(10)
return res.resnet20(100)
elif modelType=="resnet44":
if dataset == "MNIST":
print("MNIST dataset not supported in this model. Try resnet20 or 32")
assert (False)
return res.resnet44(100)
elif modelType=="test":
if dataset=="MNIST":
print ("MNIST dataset not supported in this model. Try resnet20 or 32")
assert(False)
return tm.Net(100)
else:
print ("Unsupported model; either implement the model in model/modelFactory or choose a different model")
assert(False)
def get_model(model_type, dataset="CIFAR100"):
if model_type == "resnet32":
if dataset == "MNIST":
return res.resnet32mnist(10)
elif dataset == "CIFAR10":
return res.resnet32(10)
return res.resnet32(100)
elif model_type == "resnet20":
if dataset == "MNIST":
return res.resnet20mnist(10)
elif dataset == "CIFAR10":
return res.resnet20(10)
return res.resnet20(100)
elif model_type == "resnet10":
if dataset == "MNIST":
return res.resnet10mnist(10)
elif dataset == "CIFAR10":
return res.resnet20(10)
return res.resnet20(100)
elif model_type == "resnet44":
if dataset == "MNIST":
print(
"MNIST Dataset not supported in this model. Try resnet20 or 32"
)
assert (False)
elif dataset == "CIFAR10":
return res.resnet44(10)
return res.resnet44(100)
elif model_type == "test":
if dataset == "MNIST":
return tm.Net(10, 1)
elif dataset == "CIFAR10":
return tm.Net(10)
return tm.Net(100)
else:
print(
"Unsupported model; either implement the model in model/ModelFactory or choose a different model"
)
assert (False)
def get_model(model_type, dataset):
if model_type == "resnet":
if dataset == 'document':
return resnet.resnet20(8)
elif dataset == 'corner':
return resnet.resnet20(2)
elif model_type == "resnet8":
if dataset == 'document':
return resnet.resnet8(8)
elif dataset == 'corner':
return resnet.resnet8(2)
elif model_type == "resnet32":
if dataset == 'document':
return resnet.resnet32(8)
elif dataset == 'corner':
return resnet.resnet32(2)
elif model_type == "resnet44":
if dataset == 'document':
return resnet.resnet44(8)
elif dataset == 'corner':
return resnet.resnet44(2)
elif model_type == 'shallow':
if dataset == 'document':
return tm.MobileNet(8)
elif dataset == 'corner':
return tm.MobileNet(2)
elif model_type == "squeeze":
if dataset == 'document':
return models.squeezenet1_1(True)
elif dataset == 'corner':
return models.squeezenet1_1(True)
else:
print(
"Unsupported model; either implement the model in model/ModelFactory or choose a different model"
)
assert (False)
def get_model(self, model_type, dataset="CIFAR100", use_mbd=False, d=64):
if model_type == "densenet":
if dataset == "MNIST":
print(
"MNIST dataset not supported in this model. Try resnet20 or 32"
)
assert (False)
return dn.DenseNet(growth_rate=12,
depth=40,
reduction=0.5,
bottleneck=True,
n_classes=100)
elif model_type == "resnet32":
if dataset == "MNIST":
return res.resnet32mnist(10)
elif dataset == "CIFAR10":
return res.resnet32(10)
return res.resnet32(100)
elif model_type == "resnet20":
if dataset == "MNIST":
return res.resnet20mnist(10)
return res.resnet20(100)
elif model_type == "resnet44":
if dataset == "MNIST":
print(
"MNIST dataset not supported in this model. Try resnet20 or 32"
)
assert (False)
return res.resnet44(100)
elif model_type == "test":
if dataset == "MNIST":
print(
"MNIST dataset not supported in this model. Try resnet20 or 32"
)
assert (False)
return tm.Net(100)
elif model_type == "cdcgan":
if dataset == "CIFAR100":
G = cdcgan.Generator(d, 3, 100)
D = cdcgan.Discriminator(d, 3, 100, use_mbd)
elif dataset == "CIFAR10":
G = cdcgan.Generator(d, 3, 10)
D = cdcgan.Discriminator(d, 3, 10, use_mbd)
else:
G = cdcgan.Generator(d)
D = cdcgan.Discriminator(d, 1, 10, use_mbd)
G.init_weights(mean=0.0, std=0.02)
D.init_weights(mean=0.0, std=0.02)
return G, D
elif model_type == "dcgan":
if dataset == "CIFAR100" or dataset == "CIFAR10":
G = dcgan.Generator(d, 3)
D = dcgan.Discriminator(d, 3)
else:
G = dcgan.Generator(d)
D = dcgan.Discriminator(d)
G.init_weights(mean=0.0, std=0.02)
D.init_weights(mean=0.0, std=0.02)
return G, D
elif model_type == "wgan":
if dataset == "CIFAR100" or dataset == "CIFAR10":
G = wgan.Generator(d, 3)
D = wgan.Discriminator(d, 3)
else:
G = wgan.Generator(d)
D = wgan.Discriminator(d)
G.init_weights(mean=0.0, std=0.02)
D.init_weights(mean=0.0, std=0.02)
return G, D
elif model_type == "acgan":
num_classes = 100 if dataset == "CIFAR100" else 10
gen_d = 384
if d < 16:
print("[!!!] d<16, You sure??")
assert False
if d == 32:
gen_d = 768
if dataset == "CIFAR100" or dataset == "CIFAR10":
G = acgan.Generator(gen_d, 3, num_classes)
D = acgan.Discriminator(d, 3, num_classes)
else:
G = acgan.Generator(gen_d, 1, num_classes)
D = acgan.Discriminator(d, 1, num_classes)
G.init_weights(mean=0.0, std=0.02)
D.init_weights(mean=0.0, std=0.02)
return G, D
else:
print(
"Unsupported model; either implement the model in model/ModelFactory or choose a different model"
)
assert (False)