Spaces:
Runtime error
Runtime error
Commit
·
2598f17
1
Parent(s):
7659f48
Update app.py
Browse files
app.py
CHANGED
@@ -42,9 +42,9 @@ def init_model():
|
|
42 |
model = load_model(model, args.load_model_path)
|
43 |
model.eval()
|
44 |
|
45 |
-
for name, parameter in model.named_parameters():
|
46 |
-
|
47 |
-
|
48 |
print(torch.cuda.max_memory_allocated() / 1024 ** 3)
|
49 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
50 |
model.to(device)
|
|
|
42 |
model = load_model(model, args.load_model_path)
|
43 |
model.eval()
|
44 |
|
45 |
+
# for name, parameter in model.named_parameters():
|
46 |
+
# print(name)
|
47 |
+
# print(parameter)
|
48 |
print(torch.cuda.max_memory_allocated() / 1024 ** 3)
|
49 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
50 |
model.to(device)
|