ImageClassification-PyTorch
ImageClassification-PyTorch copied to clipboard
我运行了您的代码,总是出现KeyError: 0,
问题出现的位置是 for data in dataloders[phase]:,我追溯了问题,最后是在batch = self.collate_fn([self.dataset[i] for i in indices])中停止的,我使用的是python3.5版本