run_inference.sh 286 Bytes
Newer Older
yangzhong's avatar
yangzhong committed
1
2
3
4
5
accelerate launch --num_processes=1 --gpu_ids="0," --main_process_port 29300 src/inference_s3diff.py \
    --de_net_path="./pretrained_weight/zhangap/S3Diff/de_net.pth" \
    --output_dir="./output" \
    --ref_path="./datasets/Set5/image_SRF_2/HR_1024/" \
    --align_method="wavelet"