https://community.esri.com/t5/arcgis-api-for-python-questions/error-when-running-unetclassifier-model/m-p/1239559#M8138 <P>I tried running a UnetClassifier model but i keep getting an error when trying to find the optimal learning rate. The error is as follows:&nbsp;</P><PRE>RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same</PRE><P>Now i found out that this means that the model is running on the gpu and my data is on the cpu. I used the prepare data function to create my data, and i can't find a option to get my data to prepare on GPU instead of CPU.&nbsp;</P><P data-unlink="true">I am running my code in a notebook in arcgis pro version 3.0.2, the dataset i'm using is&nbsp;<A href="https://geosaurus.maps.arcgis.com/home/item.html?id=08ccd71f862940bda1d8350c8cc91a47" target="_self">land_cover_classification_using_sparse_training_data</A>&nbsp; and i'm running the code below</P><P>&nbsp;</P><LI-CODE lang="python">data = prepare_data(path=output_path, batch_size=6, imagery_type='naip', dataset_type='Classified_Tiles') model = UnetClassifier(data, backbone='resnet34', ignore_classes=[0]) lr = model.lr_find()</LI-CODE><P>&nbsp;</P> Fri, 09 Dec 2022 11:05:58 GMT StanNoordman 2022-12-09T11:05:58Z https://community.esri.com/t5/arcgis-api-for-python-questions/error-when-running-unetclassifier-model/m-p/1239559#M8138 <P>I tried running a UnetClassifier model but i keep getting an error when trying to find the optimal learning rate. The error is as follows:&nbsp;</P><PRE>RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same</PRE><P>Now i found out that this means that the model is running on the gpu and my data is on the cpu. I used the prepare data function to create my data, and i can't find a option to get my data to prepare on GPU instead of CPU.&nbsp;</P><P data-unlink="true">I am running my code in a notebook in arcgis pro version 3.0.2, the dataset i'm using is&nbsp;<A href="https://geosaurus.maps.arcgis.com/home/item.html?id=08ccd71f862940bda1d8350c8cc91a47" target="_self">land_cover_classification_using_sparse_training_data</A>&nbsp; and i'm running the code below</P><P>&nbsp;</P><LI-CODE lang="python">data = prepare_data(path=output_path, batch_size=6, imagery_type='naip', dataset_type='Classified_Tiles') model = UnetClassifier(data, backbone='resnet34', ignore_classes=[0]) lr = model.lr_find()</LI-CODE><P>&nbsp;</P> Fri, 09 Dec 2022 11:05:58 GMT https://community.esri.com/t5/arcgis-api-for-python-questions/error-when-running-unetclassifier-model/m-p/1239559#M8138 StanNoordman 2022-12-09T11:05:58Z https://community.esri.com/t5/arcgis-api-for-python-questions/error-when-running-unetclassifier-model/m-p/1241509#M8169 <P>Does anyone have a solution for this problem?</P> Thu, 15 Dec 2022 12:40:34 GMT https://community.esri.com/t5/arcgis-api-for-python-questions/error-when-running-unetclassifier-model/m-p/1241509#M8169 StanNoordman 2022-12-15T12:40:34Z