Problem with training on macOS

[Hongbo-Z]

I came across a problem when I train GG-CNN on Cornell Dataset on my MacBook Pro (15-inch, 2019)

python train_ggcnn.py --description training_example --network ggcnn --dataset cornell --dataset-path Cornell_Grasping_Dataset/

INFO:root:Loading Cornell Dataset...
INFO:root:Done
INFO:root:Loading Network...
Traceback (most recent call last):
File "train_ggcnn.py", line 272, in
run()
File "train_ggcnn.py", line 229, in run
net = net.to(device)
...
raise AssertionError("Torch not compiled with CUDA enabled")

[dougsm]

It looks like the code assumes you're training on GPU. You can change

ggcnn/train_ggcnn.py

Line 228 in 0c50aa7

device = torch.device("cuda:0")

to "cpu" instead and it should work on your computer, but it will be quite slow to train.