File size: 359 Bytes
3ef0208 |
1 2 3 4 5 6 7 8 9 10 11 |
#!/bin/bash
bash ./prepare.sh
username=`whoami`
task="CAR"
CUDA_VISIBLE_DEVICES=0,1 python main.py --n_GPUs 2 --batch_size 16 --lr 1e-4 --decay 200-400-600-800 --save_models --model LAMBDANET \
--scale 40 --patch_size 48 --save LAMBDA_Q40_RES40 --n_feats 64 --data_train DIV2K --n_resblocks 40 \
--dir_data="/data/ssd/public/$username/$task/dataset" --amp
|