pytorch图片数据集加载

json读取

fp是文件用load()，返回字典格式

fp = open('./source_file/info.json')
json_data = json.load(fp)

读取的是字符串用loads()，返回字典格式

str1 = '{"name": "张三", "age": 18, "sex": "男"}'
json_data = json.loads(str1)

详细内容见：https://www.cnblogs.com/XhyTechnologyShare/p/12033690.html

txt读取

见我的博客：https://blog.csdn.net/Never__Say__No/article/details/108540812

图片加载

https://ptorch.com/news/215.html
https://blog.csdn.net/sinat_42239797/article/details/90641659

训练

在训练模型时会在前面加上：

model.train()

在测试模型时在前面使用：

model.eval()

这两个方法是针对在网络训练和测试时采用不同方式的情况，比如Batch Normalization 和 Dropout。在model(test_datasets)之前，需要加上model.eval(). 否则的话，有输入数据，即使不训练，它也会改变权值。这是model中含有batch normalization层所带来的的性质。