From a96907a49b088160a419714185010f451dc7b03d Mon Sep 17 00:00:00 2001 From: George Yiasemis Date: Sun, 20 Oct 2024 22:47:43 +0200 Subject: [PATCH] Minor fix --- direct/nn/transformers/mri_transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/direct/nn/transformers/mri_transformers.py b/direct/nn/transformers/mri_transformers.py index 65c2f06e..d1544145 100644 --- a/direct/nn/transformers/mri_transformers.py +++ b/direct/nn/transformers/mri_transformers.py @@ -625,7 +625,7 @@ def __init__( Whether to compute the output per coil. """ super().__init__() - self.transformer = VisionTransformer2D( + self.transformer = VisionTransformer3D( average_img_size=average_size, patch_size=patch_size, in_channels=COMPLEX_SIZE,