Python train.py with modelnet40.json

1

https://github.com/valeoai/LightConvPoint/blob/master/examples/README.md

Hey, i am reexperimenting with modelnet40, however,

ERROR - ModelNet40 - Failed after 0:00:00!
Traceback (most recent calls WITHOUT Sacred internals):
File “train.py”, line 84, in main
device = torch.device(_config[“device”])
TypeError: Device() received an invalid combination of arguments - got (NoneType), but expected one of:

  • (torch.device device)
    didn’t match because some of the arguments have invalid types: (NoneType)
2

It seems that _config["device"] contains a None entry, while a valid device (or device string) is needed.
You could either make sure the _config dict contains a valid device or alternatively specify the device manually e.g. via:

device = 'cuda' if torch.cuda.is_available else 'cpu'