More experimentation
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
This commit is contained in:
parent
1c501db81a
commit
83d870d2ea
1 changed files with 740 additions and 487 deletions
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Reference in a new issue