-default_task_args = {
- "sandbox": {
- "nb_epochs": 50,
- "batch_size": 25,
- "nb_train_samples": 100000,
- "nb_test_samples": 10000,
- },
- "picoclvr": {
- "nb_epochs": 25,
- "batch_size": 25,
- "nb_train_samples": 250000,
- "nb_test_samples": 10000,
- },
- "mnist": {
- "nb_epochs": 25,
- "batch_size": 10,
- "nb_train_samples": 250000,
- "nb_test_samples": 10000,
- },
- "maze": {
- "nb_epochs": 25,
- "batch_size": 5,
- "nb_train_samples": 250000,
- "nb_test_samples": 10000,
- },
- "snake": {
- "nb_epochs": 5,
- "batch_size": 25,
- "nb_train_samples": 250000,
- "nb_test_samples": 10000,
- },
- "stack": {
- "nb_epochs": 5,
- "batch_size": 25,
- "nb_train_samples": 100000,
- "nb_test_samples": 1000,
- },
- "expr": {
- "nb_epochs": 40,
- "batch_size": 25,
- "nb_train_samples": 1000000,
- "nb_test_samples": 10000,
- },
- "rpl": {
- "nb_epochs": 40,
- "batch_size": 25,
- "nb_train_samples": 100000,
- "nb_test_samples": 10000,
- },
- "world": {
- "nb_epochs": 10,
- "batch_size": 25,
- "nb_train_samples": 25000,
- "nb_test_samples": 1000,
- },
+if args.dirty_debug:
+ args.accuracy_to_make_c_quizzes = 0.0
+ nb_new_c_quizzes_for_train = 100
+ nb_new_c_quizzes_for_test = 10
+
+######################################################################
+
+default_args = {
+ "model": "37M",
+ "batch_size": 100,
+ "nb_train_samples": 100000,
+ "nb_test_samples": 10000,