bash ./prepare.sh | |
username=`whoami` | |
task="CAR" | |
CUDA_VISIBLE_DEVICES=2,3 python main.py --n_GPUs 2 --batch_size 16 --lr 1e-4 --decay 200-400-600-800 --save_models --model LAMBDANET \ | |
--scale 10 --patch_size 96 --save LAMBDA_Q10_RES40_R3_P96 --n_feats 64 --data_train DIV2K --n_resblocks 40 \ | |
--dir_data="/data/ssd/public/$username/$task/dataset" --amp --recurrence 3 --loss 1*MSE | |