982865f
1
2
3
4
5
6
7
8
9
10
11
12
13
from .network import * from .common import * MODELS = { "Recce": Recce } def load_model(name="Recce"): assert name in MODELS.keys(), f"Model name can only be one of {MODELS.keys()}." print(f"Using model: '{name}'") return MODELS[name]