More experimentation

Signed-off-by: András Schmelczer <andras@schmelczer.dev>
This commit is contained in:
Andras Schmelczer 2022-04-03 21:46:35 +02:00
commit 83d870d2ea

View file

@ -6,12 +6,16 @@ from joblib import dump
from good_ai.open_s3 import LargeFile
from ..set_default_config import set_default_config_if_uninitialized
logger = logging.getLogger("models")
def save_model(
model: Union[Path, str, object], key: str, keep_last_n: Optional[int] = None
) -> int:
set_default_config_if_uninitialized()
file = LargeFile(name=key, mode="wb", keep_last_n=keep_last_n)
if isinstance(model, Path) or isinstance(model, str):