"benchmarks/benchmarking_flux.py" did not exist on "8c938fb410e79a0d04d727b68edf28e4036c0ca5"
val.sh 355 Bytes
Newer Older
Rayyyyy's avatar
Rayyyyy committed
1
2
3
4
5
6
7
8
#!/bin/bash
echo "Export params ..."
export HIP_VISIBLE_DEVICES=0 # 自行修改为训练的卡号

# 以 HAT_SRx4_ImageNet-pretrain.pth 为例:
# 7行 datasets:确认数据地址是否一致,如不一致,进行对应修改
# 70行 pretrain_network_g,可根据自己地址修改
python hat/test.py -opt options/test/HAT_SRx4_ImageNet-pretrain.yml