worldmem-baseline-evals / GeometryForcing /scripts /eval_geometry_forcing.sh
BonanDing's picture
Add isolated Minecraft and RE10K baseline evaluation suite
59630ba verified
Raw History Blame Contribute Delete
1.31 kB
echo "Running evaluation for DFoT on RealEstate10K dataset"
eval_result_dir="$output_dir/geometry_forcing_long"
checkpoint_path="checkpoints/geometry_forcing_state_dict.ckpt"
algorithm="dfot_geometry_forcing"
echo "Result directory: $result_dir"
echo "Checkpoint path: $checkpoint_path"
python -m main +name=single_image_to_long dataset=realestate10k \
algorithm=$algorithm experiment=video_generation \
@diffusion/continuous \
load=$checkpoint_path \
'experiment.tasks=[validation]' experiment.validation.data.shuffle=False experiment.test.data.shuffle=False \
dataset.context_length=1 dataset.frame_skip=1 dataset.n_frames=256 \
algorithm.tasks.prediction.keyframe_density=0.0625 \
algorithm.tasks.interpolation.max_batch_size=4 experiment.validation.batch_size=1 \
algorithm.tasks.prediction.history_guidance.name=stabilized_vanilla \
+algorithm.tasks.prediction.history_guidance.guidance_scale=4.0 \
+algorithm.tasks.prediction.history_guidance.stabilization_level=0.02 \
algorithm.tasks.interpolation.history_guidance.name=vanilla \
+algorithm.tasks.interpolation.history_guidance.guidance_scale=1.5 \
'algorithm.logging.metrics=[fvd,fid,psnr,lpips,ssim]' \
hydra.run.dir=$result_dir