Upload folder using huggingface_hub
Browse files- README.md +18 -3
- gune_amp.pt +3 -0
- metal_amp.pt +3 -0
- metal_amp_v2.pt +3 -0
README.md
CHANGED
|
@@ -1,3 +1,18 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
## Usage
|
| 2 |
+
|
| 3 |
+
```python
|
| 4 |
+
import torch
|
| 5 |
+
from model import MetalAmpNet # make sure model.py is in the working directory
|
| 6 |
+
|
| 7 |
+
# Load checkpoint
|
| 8 |
+
ckpt = torch.load("metal_amp_v2.pt", map_location="cpu")
|
| 9 |
+
|
| 10 |
+
# Initialize model
|
| 11 |
+
model = MetalAmpNet()
|
| 12 |
+
model.load_state_dict(ckpt["model"])
|
| 13 |
+
model.eval()
|
| 14 |
+
|
| 15 |
+
# Example inference
|
| 16 |
+
# x: torch.Tensor of shape [1, 1, window_size] (single channel)
|
| 17 |
+
# y = model(x)
|
| 18 |
+
```
|
gune_amp.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f79664866e8a9810c7e891f1ea5cb59b1e94e50c3370706c7322a44c2ccdfc9c
|
| 3 |
+
size 463889
|
metal_amp.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:90321fec22fc10411a79c5a9f6f505830fb4108ca75f0c104394d11e0fc63c4f
|
| 3 |
+
size 1494159
|
metal_amp_v2.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dd85ef35e92f434dd931e1ecdfebd491c32981ce7fcb4443df2b85cc5e2cbc95
|
| 3 |
+
size 6418001
|