1 files changed, 87 insertions, 0 deletions
diff --git a/cfg/bouncingballs/bb-level1-run2.json b/cfg/bouncingballs/bb-level1-run2.json
new file mode 100644
index 0000000..a257408
--- /dev/null
+++ b/cfg/bouncingballs/bb-level1-run2.json
@@ -0,0 +1,87 @@
+{
+   "model_path": "bb_baseline",
+   "datatype": "bouncingballs",
+   "dataset": "BOUNCINGBALLS_OCCLUSION",
+   "scenario": "occlusion",
+   "num_updates": 400000,
+   "max_epochs": 1000,
+   "max_updates": 800000,
+   "learning_rate": {
+    "lr": 0.0001,
+    "deacrease_lr_every": 100000,
+    "deacrease_lr_factor": 0.75
+   },
+   "blackout": {
+    "blackout_start_timestep": 10,
+    "blackout_rate": 0.45,
+    "blackout_rate_max": 0.45,
+    "blackout_increase_every": 10000,
+    "blackout_increase_rate": 0.01
+   },
+   "phases": {
+    "start_inner_loop": 40000,
+    "shufleslots_end": 3000000,
+    "entity_pretraining_phase2_end": 30000,
+    "entity_pretraining_phase1_end": 15000,
+    "background_pretraining_end": 0
+   },
+   "defaults": {
+    "num_workers": 2,
+    "prefetch_factor": 2,
+    "statistics_offset": 10,
+    "load_optimizers": false,
+    "teacher_forcing": 5,
+    "skip_frames": 1,
+    "error_dropout": 0.1,
+    "seed": 67
+   },
+   "bptt": {
+    "bptt_start_timestep": 0,
+    "bptt_steps": 3,
+    "bptt_steps_max": 3,
+    "increase_bptt_steps_every": 100000
+   },
+   "model": {
+    "level": 2,
+    "batch_size": 128,
+    "num_objects": 3,
+    "img_channels": 3,
+    "input_size": [
+     64,
+     64
+    ],
+    "latent_size": [
+     4,
+     4
+    ],
+    "gestalt_size": 12,
+    "bottleneck": "binar",
+    "position_regularizer": 0.01,
+    "time_regularizer": 0.25,
+    "encoder_regularizer": 0.333333,
+    "latent_regularizer": 0.0,
+    "inner_loop_enabled": false,
+    "latent_loss_enabled": false,
+    "encoder": {
+     "channels": 24,
+     "level1_channels": 12,
+     "num_layers": 2,
+     "reg_lambda": 1e-10
+    },
+    "predictor": {
+     "heads": 2,
+     "layers": 3,
+     "channels_multiplier": 2,
+     "reg_lambda": 1e-10,
+     "transformer_type": "shared"
+    },
+    "decoder": {
+     "channels": 24,
+     "level1_channels": 12,
+     "num_layers": 2
+    },
+    "update_module": {
+     "reg_lambda": 1e-05
+    }
+   }
+  }
+\ No newline at end of file