Spaces:
Running
Running
Delete model
Browse files- model/README.md +0 -29
- model/best_model.pth +0 -3
model/README.md
DELETED
@@ -1,29 +0,0 @@
|
|
1 |
-
# InterFuser Model Directory
|
2 |
-
|
3 |
-
## Required Files
|
4 |
-
|
5 |
-
Place your trained InterFuser model files in this directory:
|
6 |
-
|
7 |
-
1. **`interfuser_model.pth`** - The main model weights file
|
8 |
-
2. **`config.json`** (optional) - Model configuration file
|
9 |
-
|
10 |
-
## Model Format
|
11 |
-
|
12 |
-
The model should be a PyTorch state dict saved with:
|
13 |
-
```python
|
14 |
-
torch.save(model.state_dict(), 'interfuser_model.pth')
|
15 |
-
```
|
16 |
-
|
17 |
-
## Loading in Code
|
18 |
-
|
19 |
-
The model is loaded in `model_definition.py`:
|
20 |
-
```python
|
21 |
-
model = InterFuserModel()
|
22 |
-
model.load_state_dict(torch.load('model/interfuser_model.pth', map_location='cpu'))
|
23 |
-
```
|
24 |
-
|
25 |
-
## Note
|
26 |
-
|
27 |
-
- The current implementation uses a dummy model for testing
|
28 |
-
- Replace with your actual trained InterFuser weights
|
29 |
-
- Ensure the model architecture matches the one defined in `model_definition.py`
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
model/best_model.pth
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:07b0906be24961fc9e5613a52c5364c41d671e9ec218f678548cd5158fe9b256
|
3 |
-
size 545502334
|
|
|
|
|
|
|
|