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
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