Closed pvjosue closed 1 year ago
The wrong gpu ordinal was being used, based on a confusion with PytorchLighning training, also I added proper checks for cpu/gpu usage
The wrong gpu ordinal was being used, based on a confusion with PytorchLighning training, also I added proper checks for cpu/gpu usage