DataLoader trong Python

[Xuan_Hoang]

E có chạy thử 1 project nhận diện trái cây trên github , nhưng đến đoạn này

train_set = FrutDataset(path='/content', mode='train')
val_set = FrutDataset(path='/content', mode='test')

def my_collate(batch):
    image = [img[0].to(device) for img in batch]
    target = [trgt[1] for trgt in batch]
    target = [{k: v.to(device) for k, v in t.items()} for t in target]
    return image, target

batch_size = 3
train_loader = torch.utils.data.DataLoader(train_set, batch_size=batch_size, 
                                           shuffle=True, collate_fn=my_collate)
val_loader = torch.utils.data.DataLoader(val_set, batch_size=1, 
                                         shuffle=False, collate_fn=my_collate)

thì e lại gặp lỗi này:

còn đây là link của project đó :

Catolampus/Fruit_detection/blob/master/Fruit_detection_train.ipynb

Các tiền bối giúp e với ạ