params["lamda"],
fishers,
prev_opt_thetas,
epoch,
description="Train on task B")
print()
loss_testB, acc_testB = test(model,
testB_loader,
base_loss_fn,
description="Test on task B")
print()
_, acc_testA = test(model,
testA_loader,
base_loss_fn,
description="Test on task A")
print()
# Callbacks
checkpoint.backup(loss_testB)
flg_stop = earlystop.check(loss_testB)
logger.update(loss_trainB=loss_trainB,
loss_testB=loss_testB,
acc_testA=acc_testA,
acc_testB=acc_testB)
if flg_stop:
break
print("------------------------------------------------------------------")
# Visualize the training progress
logger.visualize(fg_idx=1)
