a.scherbin
commited on
Commit
·
0ed3d55
1
Parent(s):
3ed92bc
Add information about usage
Browse files
README.md
CHANGED
|
@@ -21,6 +21,11 @@ Please, keep in mind that acceleration by device latency differs from accelerati
|
|
| 21 |
| baseline | 302.39 B | 1.0 | 3.381 |
|
| 22 |
| ENOT optimized | 120.95 B | 2.5 | 3.386 |
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
|
|
|
| 25 |
|
| 26 |
If you want to book a demo, please contact us: enot@enot.ai .
|
|
|
|
| 21 |
| baseline | 302.39 B | 1.0 | 3.381 |
|
| 22 |
| ENOT optimized | 120.95 B | 2.5 | 3.386 |
|
| 23 |
|
| 24 |
+
You can use `Baseline_model.pth` and `ENOT_optimized_model.pth` in the original repo by loading them as generator in the following way:
|
| 25 |
+
```python
|
| 26 |
+
generator = torch.load("ENOT_optimized_model.pth")
|
| 27 |
+
```
|
| 28 |
|
| 29 |
+
There are model objects, saved by `torch.save`, so you can load them only from the original repository root because of imports.
|
| 30 |
|
| 31 |
If you want to book a demo, please contact us: enot@enot.ai .
|