deepke/example/ae/regular/conf/predict.yaml

2 lines
91 B
YAML

# 自定义模型存储的路径
fp = 'xxx/checkpoints/2019-12-03_17-35-30/cnn_epoch21.pth'