Open RandomForestGump opened 3 years ago
Running your sample code on CPU without cuda arguments in creating train_dl results in too many values to unpack error while training. Any solution?
maybe,you can change like this:
for j, data in enumerate(train_dl): x,y=data[0],data[1]
Running your sample code on CPU without cuda arguments in creating train_dl results in too many values to unpack error while training. Any solution?