Update Berzelius training script overrides
Browse files
train_dememwm_full_berzelius.sh
CHANGED
|
@@ -29,7 +29,7 @@ srun python -m main \
|
|
| 29 |
+name=train_dememwm_full_berzelius_8a100_bs8_global64_350k \
|
| 30 |
+output_dir="${OUTPUT_DIR}/" \
|
| 31 |
auto_resume=true \
|
| 32 |
-
|
| 33 |
algorithm=dememwm_memory_dit \
|
| 34 |
+customized_load=true \
|
| 35 |
+seperate_load=true \
|
|
@@ -47,7 +47,7 @@ srun python -m main \
|
|
| 47 |
+dataset.angle_range=180 \
|
| 48 |
+dataset.pos_range=8 \
|
| 49 |
++algorithm.n_tokens=4 \
|
| 50 |
-
|
| 51 |
++algorithm.context_frames=100 \
|
| 52 |
++algorithm.log_video=true \
|
| 53 |
++algorithm.diffusion.sampling_timesteps=20 \
|
|
|
|
| 29 |
+name=train_dememwm_full_berzelius_8a100_bs8_global64_350k \
|
| 30 |
+output_dir="${OUTPUT_DIR}/" \
|
| 31 |
auto_resume=true \
|
| 32 |
+
experiment.tasks=[training] \
|
| 33 |
algorithm=dememwm_memory_dit \
|
| 34 |
+customized_load=true \
|
| 35 |
+seperate_load=true \
|
|
|
|
| 47 |
+dataset.angle_range=180 \
|
| 48 |
+dataset.pos_range=8 \
|
| 49 |
++algorithm.n_tokens=4 \
|
| 50 |
+
algorithm.x_shape=[16,18,32] \
|
| 51 |
++algorithm.context_frames=100 \
|
| 52 |
++algorithm.log_video=true \
|
| 53 |
++algorithm.diffusion.sampling_timesteps=20 \
|