Running small dataset on CPU

hello, I am running into some issues when I attempt to run the train_unet script on a small dataset on my local environment (which does not have any GPU capabilities).

When I run the script with the --gpus argument set to 0, I get the following error message: File "/Users/abhinavsaksena/mri/fastMRI/env/lib/python3.7/site-packages/torch/utils/tensorboard/summary.py", line 156, in hparams raise ValueError('value should be one of int, float, str, bool, or torch.Tensor') ValueError: value should be one of int, float, str, bool, or torch.Tensor

If I try to change the --gpu arg to 1, I get the following message: File "/Users/abhinavsaksena/mri/fastMRI/env/lib/python3.7/site-packages/pytorch_lightning/trainer/distrib_parts.py", line 533, in sanitize_gpu_ids raise MisconfigurationException(message) pytorch_lightning.utilities.debugging.MisconfigurationException: You requested GPUs: [0] But your machine only has: []

How might I go about fixing these issues and run the code on my CPU? Here are some similar issues I found online for reference: https://github.com/PyTorchLightning/pytorch-lightning/pull/609 https://github.com/PyTorchLightning/pytorch-lightning/issues/899

[asaksena98]

I updated the pytorch-lightning version to 0.7.6 and that seemed to fix it.

[mmuckley]

Closing this as the problem is fixed.