File size: 766 Bytes
7b0dd2f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# InterFuser Model Directory

## Required Files

Place your trained InterFuser model files in this directory:

1. **`interfuser_model.pth`** - The main model weights file
2. **`config.json`** (optional) - Model configuration file

## Model Format

The model should be a PyTorch state dict saved with:
```python
torch.save(model.state_dict(), 'interfuser_model.pth')
```

## Loading in Code

The model is loaded in `model_definition.py`:
```python
model = InterFuserModel()
model.load_state_dict(torch.load('model/interfuser_model.pth', map_location='cpu'))
```

## Note

- The current implementation uses a dummy model for testing
- Replace with your actual trained InterFuser weights
- Ensure the model architecture matches the one defined in `model_definition.py`