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