def main():
processes = []
## MEDIUM
config, args, opt = configurations('mnist_medium')
args.img_size = 32
args.layer_G = [(512, 4, 1, 0), (256, 4, 2, 1), (128, 4, 2, 1),
(1, 4, 2, 1)]
args.layer_D = [(512, 4, 2, 1), (256, 4, 2, 1), (128, 4, 2, 1),
(1, 4, 1, 0)]
processes.append(mp.Process(target=subtask,\
args=(config, args, opt)))
## SMALL
config, args, opt = configurations('mnist_small')
args.img_size = 28
args.layer_G = [(256, 7, 1, 0), (128, 4, 2, 1), (1, 4, 2, 1)]
args.layer_D = [(256, 4, 2, 1), (128, 4, 2, 1), (1, 7, 1, 0)]
processes.append(mp.Process(target=subtask,\
args=(config, args, opt)))
## LARGE
config, args, opt = configurations('mnist_large')
processes.append(mp.Process(target=subtask,\
args=(config, args, opt)))
# config, args, opt = configurations('mnist_1_3')
# opt.k = 3
# processes.append(mp.Process(target=subtask,\
# args=(config, args, opt)))
# config, args, opt = configurations('mnist_1_7')
# opt.k = 7
# processes.append(mp.Process(target=subtask,\
# args=(config, args, opt)))
# config, args, opt = configurations('mnist_3_1')
# opt.k = 1
# opt.g = 3
# processes.append(mp.Process(target=subtask,\
# args=(config, args, opt)))
# config, args, opt = configurations('mnist_7_1')
# opt.g = 7
# processes.append(mp.Process(target=subtask,\
# args=(config, args, opt)))
memory = gpu_gauge()
for process in processes:
while not memory.available() > 3000:
time.sleep(5)
print('hi')
print(memory.available())
process.start()
process.join(0.1)
time.sleep(15)
def main():
config, args, opt = configurations('MODLE_MODIFY_relu')
check_directories(opt.dir_list)
trainer = Trainer(config, args, opt)
args.use_relu = False
trainer.train()
config, args, opt = configurations('MODLE_MODIFY_batchnorm')
check_directories(opt.dir_list)
trainer = Trainer(config, args, opt)
args.use_batchnorm = False
trainer.train()
def main():
config, args, opt = configurations('LSUN_basic')
check_directories(opt.dir_list)
config.datatype='lsun'
opt.save_model=True
trainer = Trainer(config, args, opt)
trainer.train()
def main():
config, args, opt = configurations('BASIC_CELEBA', 'celeba')
check_directories(opt.dir_list)
trainer = Trainer(config, args, opt)
trainer.train()
def main():
config, args, opt = configurations('BASIC_MNIST')
check_directories(opt.dir_list)
trainer = Trainer(config, args, opt)
trainer.train()
def main():
config, args, opt = configurations('BASIC_MNIST')
trainer = Trainer(config, args, opt)
trainer.train()