search

AlexHex7 / Non-local_pytorch

Implementation of Non-local Block.
Apache License 2.0
1.57k stars 278 forks source link

如果使用更大的数据集？例如CIFAR100？ #52

Open longmalongma opened 1 year ago

longmalongma commented 1 year ago

感谢你很棒的工作，如果使用更大的数据集？例如CIFAR100？能提供可以跑多个不同数据集的训练脚本吗？谢谢

AlexHex7 commented 1 year ago

@longmalongma Hi,

torchvision.datasets中提供了常见的数据集(CIFAR, VOC, Cityscapes等)。

可修改代码中的data部分:

https://github.com/AlexHex7/Non-local_pytorch/blob/57f5230abd85e4bf8ab07f3089af55c7942f053b/demo_MNIST_train.py#L9-L14

例如，修改为CIFAR100, 并设计合适输入尺寸的Network. train_data = torchvision.datasets.CIFAR100(...)