-default_task_args = {
- "byheart": {
- "model": "37M",
- "nb_epochs": 2,
- "batch_size": 25,
- "nb_train_samples": 50000,
- "nb_test_samples": 10000,
- },
- "learnop": {
- "model": "37M",
- "nb_epochs": 15,
- "batch_size": 25,
- "nb_train_samples": 50000,
- "nb_test_samples": 10000,
- },
- "guessop": {
- "model": "352M",
- "nb_epochs": 5,
- "batch_size": 25,
- "nb_train_samples": 1000000,
- "nb_test_samples": 10000,
- },
- "twotargets": {
- "model": "37M",
- "nb_epochs": 10,
- "batch_size": 25,
- "nb_train_samples": 50000,
- "nb_test_samples": 10000,
- },
- "addition": {
- "model": "352M",
- "nb_epochs": 50,
- "batch_size": 25,
- "nb_train_samples": 250000,
- "nb_test_samples": 10000,
- },
- "picoclvr": {
- "model": "37M",
- "nb_epochs": 25,
- "batch_size": 25,
- "nb_train_samples": 250000,
- "nb_test_samples": 10000,
- },
- "mnist": {
- "model": "37M",
- "nb_epochs": 25,
- "batch_size": 10,
- "nb_train_samples": 60000,
- "nb_test_samples": 10000,
- },
- "maze": {
- "model": "37M",
- "nb_epochs": 25,
- "batch_size": 5,
- "nb_train_samples": 100000,
- "nb_test_samples": 10000,
- },
- "snake": {
- "model": "37M",
- "nb_epochs": 5,
- "batch_size": 25,
- "nb_train_samples": 250000,
- "nb_test_samples": 10000,
- },
- "stack": {
- "model": "37M",
- "nb_epochs": 15,
- "batch_size": 25,
- "nb_train_samples": 100000,
- "nb_test_samples": 1000,
- },
- "expr": {
- "model": "352M",
- "nb_epochs": 25,
- "batch_size": 25,
- "nb_train_samples": 2500000,
- "nb_test_samples": 10000,
- },
- "rpl": {
- "model": "122M",
- "nb_epochs": 50,
- "batch_size": 5,
- "nb_train_samples": 1000000,
- "nb_test_samples": 10000,
- },
- "world": {
- "model": "37M",
- "nb_epochs": 10,
- "batch_size": 25,
- "nb_train_samples": 25000,
- "nb_test_samples": 1000,
- },
- "grid": {
- "model": "37M",
- "nb_epochs": 25,
- "batch_size": 25,
- "nb_train_samples": 250000,
- "nb_test_samples": 10000,
- },