If you are looking to train on a single batch, then remove your loop over your dataloader: for i, data in enumerate(train_loader, ...
確定! 回上一頁