跑PSMNet遇到的坑

https://github.com/JiaRenChang/PSMNet

finetune.py是拿来跑kitti的,main是拿来跑Scene Flow的,--no-cuda参数要设置成false,这样就可以在gpu上跑

原始的代码batch_size是12,太大了,在TrainImgLoader中调小,之前一直在main里面调小发现没效果,应该在finetune里面调

with torch.no_grad()的no_grad()有问题,现在只能选择先注销掉,不进行test

猜你喜欢

转载自www.cnblogs.com/ymjyqsx/p/8968144.html