######################################################################
if args.check:
- args.nb_train_samples = 500
+ args.nb_train_samples = 2500
args.nb_test_samples = 100
if args.physical_batch_size is None:
nb_new_quizzes_for_test = 10
for n_epoch in range(args.nb_epochs):
- a = [(model.id, model.main_test_accuracy.item()) for model in models]
+ a = [(model.id, model.main_test_accuracy) for model in models]
a.sort(key=lambda p: p[0])
log_string(f"current accuracies {a}")