Closed robmarkcole closed 11 months ago
Following https://github.com/allenai/satlas/issues/13 I note a couple of minor improvements
from osgeo import gdal
device = torch.device("cuda")
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
Thanks, made these changes (updated the device in the other scripts as well)
Following https://github.com/allenai/satlas/issues/13 I note a couple of minor improvements
from osgeo import gdal
is not required for this example and can be removed (somehow this always gives headaches on mac)device = torch.device("cuda")
-> preferable to usedevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")
to avoid errors on CPU