lambdanet / DN_RGB /code /train-l.sh
hyliu's picture
Upload folder using huggingface_hub
2541ce6 verified
#!/bin/bash
bash ./prepare.sh
CUDA_VISIBLE_DEVICES=1 python main.py --n_GPUs 1 --lr 1e-5 --batch_size 16 --save_models --epoch 100 --model LAMBDANET \
--scale 50 --patch_size 128 --reset --save MDSR_lambda_N50 --n_feats 64 --data_train DIV2K --dir_data="/data/ssd/public/liuhy/sr/dataset"