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:

torch.save(model.state_dict(), 'interfuser_model.pth')

Loading in Code

The model is loaded in model_definition.py:

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