File size: 254 Bytes
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]