12 lines
192 B
Bash
12 lines
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
|