File size: 318 Bytes
3ef0208
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
#!/bin/bash

bash ./prepare.sh

username=`whoami`
task="CAR"

python main.py --n_GPUs 4 --batch_size 16 --lr 1e-4 --decay 200-400-600-800 --save_models --model RECURSIONNET \
--scale 10 --patch_size 8 --save RECURSIONNET_Q10 --n_feats 64 --data_train DIV2K \
--dir_data="/data/ssd/public/$username/$task/dataset" --amp