search

somepago / saint

The official PyTorch implementation of recent paper - SAINT: Improved Neural Networks for Tabular Data via Row Attention and Contrastive Pre-Training
Apache License 2.0
402 stars 63 forks source link

enumerate DataLoader is too slow. #20

Open leeejeonghwa opened 1 year ago

leeejeonghwa commented 1 year ago

for i, data in enumerate(trainloader, 0) #181 This code is stopped without turning around. I think it's going on an infinite loop, can you solve it?

somepago commented 1 year ago

Hi, this is just data loading step (very typical in pytorch). You could test your trainloader. Can you print "i" and see if the code is even running?

leeejeonghwa commented 1 year ago

problem Is this the right way to do it? I don't think it's progressing after print(trainloader).