From b95c1f9e317a623de904576b48fccef3325e561b Mon Sep 17 00:00:00 2001 From: cmoyates Date: Sun, 1 Mar 2026 09:46:31 -0330 Subject: [PATCH] fix: pass img_size to GreenFormer + gitignore .DS_Store Co-Authored-By: Claude Opus 4.6 --- .gitignore | 3 +++ scripts/dump_pytorch_reference.py | 4 ++-- scripts/infer_pytorch.py | 2 +- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index beb7476..8b14032 100644 --- a/.gitignore +++ b/.gitignore @@ -20,3 +20,6 @@ samples/ # Inference outputs output/ + +# OS files +.DS_Store diff --git a/scripts/dump_pytorch_reference.py b/scripts/dump_pytorch_reference.py index f91b70a..8fbbc25 100644 --- a/scripts/dump_pytorch_reference.py +++ b/scripts/dump_pytorch_reference.py @@ -158,13 +158,13 @@ class CNNRefinerModule(nn.Module): class GreenFormer(nn.Module): """Top-level CorridorKey model: Hiera encoder + dual decoder heads + CNN refiner.""" - def __init__(self) -> None: + def __init__(self, img_size: int = IMG_SIZE) -> None: super().__init__() self.encoder = timm.create_model( HIERA_MODEL_NAME, pretrained=False, features_only=True, - img_size=IMG_SIZE, + img_size=img_size, ) self._patch_first_conv() diff --git a/scripts/infer_pytorch.py b/scripts/infer_pytorch.py index a126364..b50b5a7 100644 --- a/scripts/infer_pytorch.py +++ b/scripts/infer_pytorch.py @@ -63,7 +63,7 @@ def main() -> None: torch.set_grad_enabled(False) print(f"Building model (img_size={args.img_size})...") - model = GreenFormer() + model = GreenFormer(img_size=args.img_size) model.eval() print(f"Loading checkpoint: {args.checkpoint}")