代码疑问

[wyz22966]

(1)MVTec-AD数据集在'./training/MVTec-AD/mvtec_anomaly_detection/'，但finetune_autoencoder.py 中 data_path='./data/mvtecad'，运行过程中会显示对象为空报错。我把它改为data_path='./training/MVTec-AD/mvtec_anomaly_detection/'就可以运行了。
(2)请问代码运行的轮数在哪里修改

[lewandofskee]

(1) Yes, data_path is your mvtec dataset path.
(2) You can add max_epochs parameter in pl.Trainer to set specific training epoch.
e.g. trainer = pl.Trainer(gpus=1, max_epochs=1000, precision=32,....)

[wyz22966]

thank you very much