| if [ ! -d "/data/ssd/public/liuhy/sr/dataset" ]; then | |
| workplace=`pwd` | |
| echo "Copying dataset to ssd" | |
| cd /research/dept7/liuhy/project/lambda/sr/dataset | |
| mkdir -p /data/ssd/public/liuhy/sr/dataset | |
| cp *.tar /data/ssd/public/liuhy/sr/dataset | |
| cd /data/ssd/public/liuhy/sr/dataset | |
| echo "Dumping tar in data path:" `pwd` | |
| for f in *.tar; do tar xf "$f"; done | |
| cd "$workplace" | |
| echo "Workplace:" `pwd` | |
| fi | |
| python main.py --n_GPUs 4 --rgb_range 1 --reset --save_models --lr 1e-4 --decay 200-400-600-800 --chop \ | |
| --save_results --n_resblocks 32 --n_feats 256 --res_scale 0.1 --batch_size 16 --model LAMBDAEDSR --scale 2 \ | |
| --patch_size 96 --save EDSR_LAMBDA_x2 --data_train DIV2K --dir_data="/data/ssd/public/liuhy/sr/dataset" | |