Baseer_Server / model /README.md
Adam
Deploy Baseer Self-Driving API v1.0
7b0dd2f
|
raw
history blame
766 Bytes
# 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`