convmae专题

ConvMAE实战:使用ConvMAE实现对植物幼苗的分类(非官方)(二)

训练 完成上面的步骤后,就开始train脚本的编写,新建train.py. 导入项目使用的库 import jsonimport osimport shutilimport matplotlib.pyplot as pltimport torchimport torch.nn.parallelimport torch.optim as optimimport torch.uti