16 lines
237 B
Bash
16 lines
237 B
Bash
#!/bin/bash
|
|
|
|
python -m train \
|
|
--gpu_id 2 \
|
|
--seed 123 \
|
|
--logger \
|
|
--non_blocking \
|
|
--pin_memory \
|
|
--batch_size 64 \
|
|
--num_workers 16 \
|
|
--num_epoch 300 \
|
|
--lr 5e-4 \
|
|
--dropout 0.1 \
|
|
--model_type tom_cm \
|
|
--aggr no_tom \
|
|
--hidden_dim 64
|