got an unexpected keyword argument ‘datasets‘的解决方法

发生异常: TypeError (note: full exception trace is shown but execution is paused at: )
init() got an unexpected keyword argument ‘datasets’

查看loader中自己写的dataset，是不是多写了一个s

train_loader = Data.DataLoader(
    datasets= train_data,
    batch_size=BATCH_SIZE,
    shuffle=True,
    num_workers=2    
)

j

将上述代码修改为:

train_loader = Data.DataLoader(
    dataset= train_data,
    batch_size=BATCH_SIZE,
    shuffle=True,
    num_workers=2    
)