Переместил interpolator.py внутрь src | добавил пресеты | добавил новые модели
This commit is contained in:
24
src/config/presets.py
Normal file
24
src/config/presets.py
Normal file
@@ -0,0 +1,24 @@
|
||||
from pathlib import Path
|
||||
from dataclasses import dataclass
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class Preset:
|
||||
config: Path
|
||||
checkpoint: Path
|
||||
|
||||
|
||||
SMALL = Preset(
|
||||
config=Path("src/config/AMT-S.yaml"),
|
||||
checkpoint=Path("src/pretrained/amt-s.pth"),
|
||||
)
|
||||
|
||||
LARGE = Preset(
|
||||
config=Path("src/config/AMT-L.yaml"),
|
||||
checkpoint=Path("src/pretrained/amt-l.pth"),
|
||||
)
|
||||
|
||||
GLOBAL = Preset(
|
||||
config=Path("src/config/AMT-g.yaml"),
|
||||
checkpoint=Path("src/pretrained/amt-g.pth"),
|
||||
)
|
||||
Reference in New Issue
Block a user