counterfactual-world-models / cwm /eval /Physion /run_eval_mp4s_keyp_flow.sh
rahulvenkk
app.py updated
6dfcb0f
raw
history blame contribute delete
970 Bytes
#physion_feature_extract \
#--model_path /ccn2/u/honglinc/cwm_checkpoints/ablation_3frame_no_clumping_mr0.90_extra_data_ep400/checkpoint-399.pth \
#--data_root_path /ccn2/u/rmvenkat/data/physion_mp4s/ \
#--model_class feature_extractor_cleaned.CWM_base_8x8_3frame_keypoints_flow \
#--gpu 7 \
#--batch_size 8 \
#--dir_for_saving /ccn2/u/rmvenkat/data/physion_release_keyp_flow/ \
#--mode ocp
physion_train_readout \
--train-path /ccn2/u/rmvenkat/data/physion_release_keyp_flow/ocp/train_features.hdf5 \
--test-path /ccn2/u/rmvenkat/data/physion_release_keyp_flow/ocp/test_features.hdf5 \
--model-name CWM_base_8x8_3frame \
--train-scenario-indices /ccn2/u/rmvenkat/data/physion_release_keyp_flow/ocp/train_json.json \
--test-scenario-indices /ccn2/u/rmvenkat/data/physion_release_keyp_flow/ocp/test_json.json \
--test-scenario-map /ccn2/u/rmvenkat/data/physion_release_keyp_flow/ocp/test_scenario_map.json \
--save_path /ccn2/u/rmvenkat/data/physion_release_keyp_flow/