def eval(self):
self.learn_sharedmodel()
model_utils.calculateFisherMatrix(self)
model_utils.UpdateMultiTaskWeightWithAlphas(self.global_index, self.modelMain)
self.modelMaintainMode = Model_Structure()
model_utils.CalculateModeParametes(self)
model_utils.CalculateModePerClassParametes(self)
# pdb.set_trace()
for label_test in range(self.opt.num_class):
self.learn_sharedmodel_w(label_test)
self.modelMain[label_test].eval()
self.modelMain[label_test].eval_flag = True
print("============== Train task #%d (Mean-IMM) ==============" %
no_of_task)
LW = model_utils.UpdateMultiTaskLwWithAlphas(L_copy[0], alpha_list,
no_of_task)
model_utils.AddMultiTaskLayers(sess, L_copy, mlp.Layers, LW,
no_of_task)
ret = mlp.TestTasks(sess, x, y, x_, y_, debug=False)
utils.PrintResults(alpha, ret)
mlp.TestAllTasks(sess, x_, y_)
######################### Mode-IMM ##########################
if mode_imm:
print("")
print("Main experiment on Drop-transfer + Mode-IMM, shuffled MNIST")
print("============== Train task #%d (Mode-IMM) ==============" %
no_of_task)
LW = model_utils.UpdateMultiTaskWeightWithAlphas(
FM, alpha_list, no_of_task)
model_utils.AddMultiTaskLayers(sess, L_copy, mlp.Layers, LW,
no_of_task)
ret = mlp.TestTasks(sess, x, y, x_, y_, debug=False)
utils.PrintResults(alpha, ret)
mlp.TestAllTasks(sess, x_, y_)
print("")
print("Time: %.4f s" % (time.time() - start))