mtomnet/tbd/run_test.sh
2025-01-10 15:39:20 +01:00

12 lines
No EOL
192 B
Bash

#!/bin/bash
python -m test \
--gpu_id 1 \
--seed 1 \
--non_blocking \
--pin_memory \
--model_type tom_cm \
--aggr no_tom \
--hidden_dim 64 \
--batch_size 64 \
--load_model_path /PATH/TO/model