rahulvenkk
app.py updated
6dfcb0f
raw
history blame
682 Bytes
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