| bash ./prepare.sh | |
| lst=0,1 | |
| if nvidia-smi -i 0 | grep -q "python"; then | |
| lst=2,3 | |
| fi | |
| CUDA_VISIBLE_DEVICES=$lst python main.py --n_GPUs 2 --lr 1e-4 --decay 200-400-600-800 --epoch 1000 --batch_size 16 --n_resblocks 10 --save_models \ | |
| --model RAFTNETLAYER --scale 1 --patch_size 48 --save RAFTS_layer_DEMOSAIC20_R4 --n_feats 64 --data_train DIV2K --recurrence 5 --data_range "1-800/901-942" \ | |
| --normalization instance --gclip 1.2 | |
| # python main.py --model LAMBDANET --n_resblocks 20 --recurrence 1 --save_results --n_GPUs 1 --chop --data_test McM+Kodak24+CBSD68+Urban100 --scale 1 \ | |
| # --pre_train ../experiment/LAMBDA_DEMOSAIC20_R1/model/model_best.pt --test_only | |