Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
dir_for_saving=/ccn2/u/rmvenkat/data/physion_release/ | |
model_name=CWM_base_8x8_3frame | |
physion_feature_extract \ | |
--data_root_path /ccn2/u/rmvenkat/data/download_test/physion_mp4s/ \ | |
--model_class feature_extractor.$model_name \ | |
--gpu 1 \ | |
--batch_size 8 \ | |
--dir_for_saving $dir_for_saving \ | |
--mode ocd | |
physion_train_readout \ | |
--train-path ${dir_for_saving}ocd/train_features.hdf5 \ | |
--test-path ${dir_for_saving}ocd/test_features.hdf5 \ | |
--model-name $model_name \ | |
--train-scenario-indices ${dir_for_saving}ocd/train_json.json \ | |
--test-scenario-indices ${dir_for_saving}ocd/test_json.json \ | |
--test-scenario-map ${dir_for_saving}ocd/test_scenario_map.json \ | |
--save_path $dir_for_saving | |