diff --git a/examples/1_SimpleNet/pt2ts.py b/examples/1_SimpleNet/pt2ts.py index 58755dde..35c22350 100644 --- a/examples/1_SimpleNet/pt2ts.py +++ b/examples/1_SimpleNet/pt2ts.py @@ -1,7 +1,6 @@ """Load a PyTorch model and convert it to TorchScript.""" import os -import sys from typing import Optional # FPTLIB-TODO diff --git a/examples/1_SimpleNet/simplenet_infer_python.py b/examples/1_SimpleNet/simplenet_infer_python.py index d02714e5..2851cef9 100644 --- a/examples/1_SimpleNet/simplenet_infer_python.py +++ b/examples/1_SimpleNet/simplenet_infer_python.py @@ -1,7 +1,6 @@ """Load saved SimpleNet to TorchScript and run inference example.""" import os -import sys import torch diff --git a/examples/2_ResNet18/pt2ts.py b/examples/2_ResNet18/pt2ts.py index 0e5db241..2db59bfb 100644 --- a/examples/2_ResNet18/pt2ts.py +++ b/examples/2_ResNet18/pt2ts.py @@ -1,7 +1,6 @@ """Load a PyTorch model and convert it to TorchScript.""" import os -import sys from typing import Optional # FPTLIB-TODO diff --git a/examples/2_ResNet18/resnet_infer_python.py b/examples/2_ResNet18/resnet_infer_python.py index 43f9e410..1834bb17 100644 --- a/examples/2_ResNet18/resnet_infer_python.py +++ b/examples/2_ResNet18/resnet_infer_python.py @@ -1,7 +1,6 @@ """Load ResNet-18 saved to TorchScript and run inference with an example image.""" import os -import sys from math import isclose import numpy as np diff --git a/examples/3_MultiGPU/multigpu_infer_fortran.f90 b/examples/3_MultiGPU/multigpu_infer_fortran.f90 index d2ad95ce..d5d49980 100644 --- a/examples/3_MultiGPU/multigpu_infer_fortran.f90 +++ b/examples/3_MultiGPU/multigpu_infer_fortran.f90 @@ -87,7 +87,7 @@ program inference 200 format("output on device ", i1,": [", 4(f5.1,","), f5.1,"]") ! Check output tensor matches expected value - expected = [(device_index + 2*i, i = 0, 4)] + expected = [(2 * (device_index + i), i = 0, 4)] test_pass = assert_allclose(out_data, expected, test_name="MultiGPU") ! Cleanup diff --git a/examples/3_MultiGPU/multigpu_infer_python.py b/examples/3_MultiGPU/multigpu_infer_python.py index 55b41411..1bc5f8f1 100644 --- a/examples/3_MultiGPU/multigpu_infer_python.py +++ b/examples/3_MultiGPU/multigpu_infer_python.py @@ -1,5 +1,7 @@ """Load saved SimpleNet to TorchScript and run inference example.""" +import os + import torch diff --git a/examples/3_MultiGPU/pt2ts.py b/examples/3_MultiGPU/pt2ts.py index 6036cb24..823e69dd 100644 --- a/examples/3_MultiGPU/pt2ts.py +++ b/examples/3_MultiGPU/pt2ts.py @@ -1,7 +1,6 @@ """Load a PyTorch model and convert it to TorchScript.""" import os -import sys from typing import Optional # FPTLIB-TODO diff --git a/examples/3_MultiGPU/simplenet.py b/examples/3_MultiGPU/simplenet.py index 7d951e12..46f46550 100644 --- a/examples/3_MultiGPU/simplenet.py +++ b/examples/3_MultiGPU/simplenet.py @@ -68,7 +68,7 @@ def forward(self, batch: torch.Tensor) -> torch.Tensor: f" {input_tensor_gpu.get_device()}" ) with torch.no_grad(): - output_tensor = model(input_tensor_gpu) + output_tensor = model(input_tensor_gpu).to("cpu") print(output_tensor) if not torch.allclose(output_tensor, 2 * input_tensor): diff --git a/examples/4_MultiIO/multiionet_infer_python.py b/examples/4_MultiIO/multiionet_infer_python.py index 12ad86a8..a615289c 100644 --- a/examples/4_MultiIO/multiionet_infer_python.py +++ b/examples/4_MultiIO/multiionet_infer_python.py @@ -1,7 +1,6 @@ """Load saved MultIONet to TorchScript and run inference example.""" import os -import sys import torch diff --git a/examples/4_MultiIO/pt2ts.py b/examples/4_MultiIO/pt2ts.py index 8f81f7f7..31066406 100644 --- a/examples/4_MultiIO/pt2ts.py +++ b/examples/4_MultiIO/pt2ts.py @@ -1,7 +1,6 @@ """Load a PyTorch model and convert it to TorchScript.""" import os -import sys from typing import Optional # FPTLIB-TODO diff --git a/examples/5_Looping/pt2ts.py b/examples/5_Looping/pt2ts.py index 5dbdad2f..962903fd 100644 --- a/examples/5_Looping/pt2ts.py +++ b/examples/5_Looping/pt2ts.py @@ -1,7 +1,6 @@ """Load a PyTorch model and convert it to TorchScript.""" import os -import sys from typing import Optional # FPTLIB-TODO diff --git a/utils/pt2ts.py b/utils/pt2ts.py index fa23f4fd..dad047c5 100644 --- a/utils/pt2ts.py +++ b/utils/pt2ts.py @@ -1,7 +1,6 @@ """Load a PyTorch model and convert it to TorchScript.""" import os -import sys from typing import Optional # FPTLIB-TODO