/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/interface/pretrained.py
16 строк
333 B
Hongxin Liu
[lazy] support from_pretrained (#4801)
26 сен 2023, 06:04
Не верифицирован
26 сен 2023, 06:04
4965c0d
Код
Авторство
О чём код?
from typing import Optional from torch.nn import Module __all__ = [ "get_pretrained_path", "set_pretrained_path", ] def get_pretrained_path(model: Module) -> Optional[str]: return getattr(model, "_pretrained", None) def set_pretrained_path(model: Module, path: str) -> None: setattr(model, "_pretrained", path)