|
wandb disabled |
|
|
|
export WANDB_API_KEY=a4f0741e80f509317597ad944a7292fabcb68bdf |
|
|
|
CHECKPOINT_PATH="checkpoints/diffusion_only.ckpt" |
|
|
|
python -m app +name=pumpkin \ |
|
algorithm=df_video_worldmemminecraft \ |
|
+checkpoint_path=$CHECKPOINT_PATH \ |
|
experiment.tasks=[interactive] \ |
|
dataset.validation_multiplier=1 \ |
|
dataset=video_minecraft \ |
|
+customized_load=true \ |
|
+dataset.n_frames_valid=100 \ |
|
+algorithm.n_tokens=8 \ |
|
+load_vae=false \ |
|
+load_t_to_r=false \ |
|
+zero_init_gate=false \ |
|
experiment.validation.batch_size=1 \ |
|
+algorithm.pose_cond_dim=5 \ |
|
+algorithm.condition_similar_length=8 \ |
|
+dataset.condition_similar_length=8 \ |
|
+algorithm.use_plucker=true \ |
|
+dataset.use_plucker=true \ |
|
+dataset.padding_pool=10 \ |
|
+dataset.focal_length=0.35 \ |
|
+algorithm.focal_length=0.35 \ |
|
+only_tune_refer=false \ |
|
+dataset.customized_validation=true \ |
|
+algorithm.customized_validation=true \ |
|
algorithm.context_frames=90 \ |
|
+algorithm.vis_gt=true \ |
|
+algorithm.relative_embedding=true \ |
|
dataset.save_dir=data/test_pumpkin \ |
|
+algorithm.log_video=true \ |
|
experiment.training.data.num_workers=4 \ |
|
experiment.validation.data.num_workers=4 \ |
|
+dataset.angle_range=30 \ |
|
+dataset.pos_range=0.5 \ |
|
+algorithm.cond_only_on_qk=true \ |
|
+algorithm.add_pose_embed=false \ |
|
+algorithm.use_domain_adapter=false \ |
|
+algorithm.use_reference_attention=true \ |
|
+algorithm.add_frame_timestep_embedder=true \ |
|
+dataset.add_frame_timestep_embedder=true \ |
|
experiment.validation.limit_batch=1 \ |
|
algorithm.diffusion.sampling_timesteps=20 \ |
|
+algorithm.is_interactive=true \ |
|
+vae_path=checkpoints/vae_only.ckpt \ |
|
+pose_predictor_path=checkpoints/pose_prediction_model_only.ckpt |
|
|
