ff345c2e2210061141653a19ef1431c6866cbf68,examples/pytorch/pointcloud/pointnet/train_partseg.py,,,#,146

Before Change

    train(net, opt, scheduler, train_loader, dev)
    if (epoch + 1) % 5 == 0:
        print("Epoch //%d Testing" % epoch)
        test_miou, test_per_cat_miou = evaluate(net, test_loader, dev, (epoch + 1) % 5 ==0)
        if test_miou > best_test_miou:
            best_test_miou = test_miou
            best_test_per_cat_miou = test_per_cat_miou
            if args.save_model_path:

After Change

// dev = "cpu"
if args.model == "pointnet":
    net = PointNetPartSeg(50, 3, 2048)
elif args.model == "pointnet2_ssg":
    net = PointNet2SSGPartSeg(50, batch_size, input_dims=6)
elif args.model == "pointnet2_msg":
    net = PointNet2MSGPartSeg(50, batch_size, input_dims=6)

net = net.to(dev)
if args.load_model_path:
    net.load_state_dict(torch.load(args.load_model_path, map_location=dev))

Instances