Minor fix

NKI-AI · Oct 20, 2024 · 8d7cd07 · 8d7cd07
1 parent 842dd87
commit 8d7cd07
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/direct/nn/transformers/config.py b/direct/nn/transformers/config.py
@@ -42,9 +42,11 @@ class ImageDomainMRIViT3DConfig(MRIViTConfig):
 class KSpaceDomainMRIViT2DConfig(MRIViTConfig):
     average_size: tuple[int, int] = (320, 320)
     patch_size: tuple[int, int] = (16, 16)
+    compute_per_coil: bool = True
 
 
 @dataclass
 class KSpaceDomainMRIViT3DConfig(MRIViTConfig):
     average_size: tuple[int, int] = (320, 320, 320)
     patch_size: tuple[int, int] = (16, 16, 16)
+    compute_per_coil: bool = True