SpringRainawa commited on
Commit
4543306
·
verified ·
1 Parent(s): 914158b

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +27 -3
README.md CHANGED
@@ -1,3 +1,27 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # BraMT on TUEG Checkpoint
2
+
3
+ This is a checkpoint of BraMT trained on the TUEG dataset using Huawei NPU.
4
+
5
+ ## Details
6
+
7
+ - **Model**: BraMT
8
+ - **Training Dataset**: TUEG
9
+ - **Hardware**: Huawei NPU
10
+
11
+ ## GitHub Repository
12
+
13
+ The full source code and training scripts are available at:
14
+ https://github.com/springrain-i/BraMT/tree/npu
15
+
16
+ ## Usage
17
+
18
+ ```python
19
+ import torch
20
+
21
+ # Load the checkpoint
22
+ checkpoint = torch.load("BraMT_on_TUEG.pth")
23
+
24
+ # Use the model
25
+ # model = YourModelClass()
26
+ # model.load_state_dict(checkpoint)
27
+ ```