# del model
# model = model_
# model.net2net_deeper_nononline()
# run_training(model, 'WiderDeeper_student_', args.epochs, plot)
# print(" >> Time tkaen by whole net2net training {}".format(time.time() - start_t))
# wider teacher training
start_t = time.time()
print("\n\n > Wider teacher training ... ")
colors.append('green')
trace_names.extend(['Wider Random Train', 'Wider Random Test'])
model_ = ConvNet(net_dataset=CIFAR10)
del model
model = model_
model.define_wider(widening_factor=2)
model.cuda()
print model
_, log_random_init = run_training(model, 'Wider_teacher_', args.epochs)
print(" >> Time taken {}".format(time.time() - start_t))
logs.append(log_random_init)
# # wider deeper teacher training
# print("\n\n > Wider+Deeper teacher training ... ")
# start_t = time.time()
# model_ = Net()
#
# del model
# model = model_
# model.define_wider_deeper()
# run_training(model, 'Wider_Deeper_teacher_', args.epochs + 1)
