SR/code/train.sh

#!/bin/sh

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"