sgune commited on
Commit
d26b604
·
verified ·
1 Parent(s): 62dc18a

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. README.md +18 -3
  2. gune_amp.pt +3 -0
  3. metal_amp.pt +3 -0
  4. metal_amp_v2.pt +3 -0
README.md CHANGED
@@ -1,3 +1,18 @@
1
- ---
2
- license: gpl-3.0
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