Pointnet_Pointnet2_pytorch
Pointnet_Pointnet2_pytorch copied to clipboard
Stuck in classifier = MODEL.get_model(num_part, normal_channel=args.normal).cuda()
Excuse me
When I try to run train_partseg.py with default parameters, I am stuck in classifier = MODEL.get_model(num_part, normal_channel=args.normal).cuda(). What could possibly be the reason? CUDA version is wrong?