weixin_48456383的博客

不要用pytorch-fid，太老了，而且权重文件有问题，你会报EOF.另外不要用pytorch-fid-helper,好像对pytorch部分版本不兼容。原项目地址:https://github.com/toshas/torch-fidelityhttps://github.com/toshas/torch-fidelity文档地址:Usage Examples with Command Line — torch-fidelity 0.3.0 documentationhttps://torc

导入数据集import torchimport torch.nn as nnimport torch.optim as optimfrom torchvision import datasetsfrom torchvision import transformsimport torch.nn.functional as Fimport torch.utils.data as Dataimport matplotlib.pyplot as pltimport numpy as npim

1.1 list 转 numpyndarray = np.array(list)1.2 numpy 转 listlist = ndarray.tolist()2.1 list 转 torch.Tensortensor=torch.Tensor(list)2.2 torch.Tensor转list先转numpy，后转listlist = tensor.numpy().tolist()3.1 torch.Tensor 转 numpyndarray = ..