#!/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