File size: 764 Bytes
d6ec83b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
#!/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"
|