Spaces:
Runtime error
Runtime error
kz209
commited on
Commit
·
c5c9ee6
1
Parent(s):
5e80f43
update
Browse files- utils/model.py +1 -0
utils/model.py
CHANGED
|
@@ -40,6 +40,7 @@ class Model(torch.nn.Module):
|
|
| 40 |
|
| 41 |
logging.info(f'Loaded model {self.name}')
|
| 42 |
|
|
|
|
| 43 |
self.update()
|
| 44 |
|
| 45 |
@classmethod
|
|
|
|
| 40 |
|
| 41 |
logging.info(f'Loaded model {self.name}')
|
| 42 |
|
| 43 |
+
self.model.eval()
|
| 44 |
self.update()
|
| 45 |
|
| 46 |
@classmethod
|