pytorch
Dataset クラスを継承し__getitem__()メソッドを index を returnするようにオーバーライドする。 class SubClass(Dataset): 略 def __getitem__(self, idx): data, target = seld.data[idx] return data, target, idx みたいにして、 dataset = SubClass(..…
Dataset クラスを継承し__getitem__()メソッドを index を returnするようにオーバーライドする。 class SubClass(Dataset): 略 def __getitem__(self, idx): data, target = seld.data[idx] return data, target, idx みたいにして、 dataset = SubClass(..…