Nien-Jung Lee
Nien-Jung Lee
Thank you for your answer. I have set the batch size to 3, but it still didn't work. It really confused me that a 16GB GPU was out of memory....
I found I forgot to resize the image causing to the memory problem. However, I face another problem. Traceback (most recent call last): File "c:/Users/10806337/Desktop/PortableGit/Projects/Robust-Lane-Detection/LaneDetectionCode/train.py", line 122, in train(args, epoch,...
I have solved my problem. I accidentally add torch.no_grad(): before the training loop. But I found one bug in the code that is in train(): loss = criterion(output, target) should...
@SeokjuLee @screamdw @ycdhqzhiai same problem.
@SeokjuLee @chenchaohui same issue. Hope for help. Thanks. How to label our own dataset?
@chengm15 Hello, I was wondering if you could show us how to visualize the result with the multi-label. I don't even know how to get the classes of the lane....
Same error. Did anybody fix the problem? Does gtop work on DRIVE Xavier?