Skip to content

Commit 503b4ed

Browse files
Can-ZhaorootKumoLiu
authored
fix bug in maisi vae (#8517)
Fixes # . ### Description A few sentences describing the changes proposed in this pull request. ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. Signed-off-by: root <[email protected]> Co-authored-by: root <[email protected]> Co-authored-by: YunLiu <[email protected]>
1 parent 8ee3f89 commit 503b4ed

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/apps/generation/maisi/networks/autoencoderkl_maisi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
232232
if self.print_info:
233233
logger.info(f"Number of splits: {self.num_splits}")
234234

235-
if self.dim_split <= 1 and self.num_splits <= 1:
235+
if self.num_splits <= 1:
236236
x = self.conv(x)
237237
return x
238238

0 commit comments

Comments
 (0)