Spaces:
Sleeping
Sleeping
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
|
|
|
|
|
|
|
|
|
|
|
|