Module


MODULECLASStorch.nn.Module(args, kwargs*)[SOURCE]层、模型的父类Base class for all neural network modules.Your models should also subclass this class.Modules can also contain other Modules, allowing to nest them in a tree structure. You can assign the submodul...

Dataset&DataLoader


dataset需要实现def __getitem__(self, index) -> T_co: # 基于一个索引返回一个训练样本(x, y)构成的训练对 raise NotImplementedError("Subclasses of Dataset should implement __getitem__.") def __len__(self): # 将数据放进去,返回数据总大小 return len(self.img_labels) re...

召唤看板娘