unknown mistake

[x12901]

Hi. I am glad to see your answer.
xiahaifeng1995/PaDiM-Anomaly-Detection-Localization-master#8
I tried to run the program but got an error. This code does not seem to be in other projects.
https://github.com/xiahaifeng1995/PaDiM-Anomaly-Detection-Localization-master
I suspect it is a memory problem. But I changed batch_size and it didn't work.

Loaded pretrained weights for efficientnet-b0
training:  None
train set feature file not exists: D:/PaDiM-EfficientNet-master/mvtec_result\model_pkl_efficientnet-b0\train_bottle.pkl
| feature extraction | test | bottle |: 100%|██████████| 1/1 [00:04<00:00,  4.65s/it]
Traceback (most recent call last):
  File "D:/PaDiM-EfficientNet-master/main.py", line 413, in <module>
    main()
  File "D:/PaDiM-EfficientNet-master/main.py", line 330, in main
    mean = torch.Tensor(train_outputs[0]).to(device)
KeyError: 0

[ingbeeedd]

Hello. It's because you don't have a pkl file because didn't train. It says training= None in output. So please add --training parameters. Please refer to usage in README because just committed. Thank you !