projects
/
culture.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[culture.git]
/
main.py
diff --git
a/main.py
b/main.py
index
09ae823
..
3b29d01
100755
(executable)
--- a/
main.py
+++ b/
main.py
@@
-183,7
+183,7
@@
for n in vars(args):
######################################################################
if args.check:
######################################################################
if args.check:
- args.nb_train_samples = 500
+ args.nb_train_samples =
2
500
args.nb_test_samples = 100
if args.physical_batch_size is None:
args.nb_test_samples = 100
if args.physical_batch_size is None:
@@
-360,7
+360,7
@@
def create_quizzes(
task.store_new_quizzes(new_quizzes[nb_for_train:], for_train=False)
task.save_image(
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,
args.result_dir,
f"world_quiz_{n_epoch:04d}_{model.id:02d}.png",
log_string,
@@
-404,7
+404,7
@@
if args.check:
nb_new_quizzes_for_test = 10
for n_epoch in range(args.nb_epochs):
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,
float(model.main_test_accuracy)
) for model in models]
a.sort(key=lambda p: p[0])
log_string(f"current accuracies {a}")
a.sort(key=lambda p: p[0])
log_string(f"current accuracies {a}")