geovit-david-beans

AbstractPhil/geovit-david-beans/
├── README.md (this file)
└── weights/
    ├── run_001_baseline_YYYYMMDD_HHMMSS/
    │   ├── best.safetensors
    │   ├── epoch_010.safetensors
    │   ├── config.json
    │   ├── training_config.json
    │   └── tensorboard/
    ├── run_002_5expert_5scale_YYYYMMDD_HHMMSS/
    │   └── ...
    └── ...

from safetensors.torch import load_file
from david_beans import DavidBeans, DavidBeansConfig
import json

# Pick a run
run_path = "weights/run_002_5expert_5scale_20251129_171229"

# Load config
with open(f"{run_path}/config.json") as f:
    config_dict = json.load(f)
config = DavidBeansConfig(**config_dict)

# Load model
model = DavidBeans(config)
state_dict = load_file(f"{run_path}/best.safetensors")
model.load_state_dict(state_dict)

# Inference
model.eval()
with torch.no_grad():
    output = model(images)
    predictions = output['logits'].argmax(dim=-1)