demo 1

app.py

@@ -4,9 +4,21 @@ import platform
 import os;
 import socket;
 import torch
+import torch.nn as nn
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+class SimpleModel(nn.Module):
+    def __init__(self):
+        super(SimpleModel, self).__init__()
+
+    def forward(self, x):
+        pass
+        
+    def device():
+       return next(model.parameters()).device
+
+model = SimpleModel().to(device);
 
 @torch.inference_mode()
 def sysinfo():
@@ -16,6 +28,7 @@ def sysinfo():
     return f"""
        hostname: {platform.node()} {socket.gethostname()}
        device: {device}
+       model device: {model.device}
        tensor: {tensorExample}
     """;