File size: 787 Bytes
913e058 622d9c9 913e058 622d9c9 c159736 622d9c9 913e058 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | export PYTHONWARNINGS="ignore"
wandb offline
python -m main +name=infer \
experiment.tasks=[validation] \
dataset.validation_multiplier=1 \
+diffusion_model_path=zeqixiao/worldmem_checkpoints/diffusion_only.ckpt \
+vae_path=zeqixiao/worldmem_checkpoints/vae_only.ckpt \
+customized_load=true \
+seperate_load=true \
dataset.n_frames=8 \
dataset.save_dir=data/minecraft \
+dataset.n_frames_valid=700 \
+dataset.memory_condition_length=8 \
+dataset.customized_validation=true \
+dataset.add_timestamp_embedding=true \
+algorithm.n_tokens=8 \
+algorithm.memory_condition_length=8 \
algorithm.context_frames=600 \
+algorithm.relative_embedding=true \
+algorithm.log_video=true \
+algorithm.add_timestamp_embedding=true
|