| export PYTHONWARNINGS="ignore" | |
| wandb offline | |
| python -m main +name=infer \ | |
| experiment.tasks=[test] \ | |
| dataset.validation_multiplier=1 \ | |
| +dataset.seed=42 \ | |
| +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 \ | |
| algorithm.diffusion.sampling_timesteps=20 \ | |
| +algorithm.memory_condition_length=8 \ | |
| +algorithm.lpips_batch_size=16 \ | |
| +algorithm.log_video=true \ | |
| +algorithm.save_local=true \ | |
| +dataset.customized_validation=true \ | |
| +algorithm.n_tokens=8 \ | |
| algorithm.context_frames=600 \ | |
| experiment.test.batch_size=1 \ | |
| experiment.test.limit_batch=10 | |