25 lines
469 B
Python
25 lines
469 B
Python
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"),
|
|
)
|