Views
No views yet
Healthy or Tumour.Healthy, Tumour)multimodal_brain_tumor_model.pth: Pretrained weights1from huggingface_hub import hf_hub_download
2import torch
3
4path = hf_hub_download("lukmanaj/brain-tumor-multimodal", "multimodal_brain_tumor_model.pth")
5model.load_state_dict(torch.load(path))| Epoch | Train Loss | Accuracy |
|---|---|---|
| 1 | 0.1552 | 94.82% |
| 5 | 0.0368 | 98.78% |
Note: The model shows signs of overfitting; further validation and augmentation are advised.