task.store_new_quizzes(new_quizzes[nb_for_train:], for_train=False)
task.save_image(
- new_quizzes[:96],
+ new_quizzes[:72],
args.result_dir,
f"world_quiz_{n_epoch:04d}_{model.id:02d}.png",
log_string,
nb_new_quizzes_for_test = 10
for n_epoch in range(args.nb_epochs):
- a = [(model.id, model.main_test_accuracy) for model in models]
+ a = [(model.id, model.main_test_accuracy.item()) for model in models]
a.sort(key=lambda p: p[0])
log_string(f"current accuracies {a}")