Skip to content

Commit d01c717

Browse files
committed
grammar
Signed-off-by: Can-Zhao <[email protected]>
1 parent f4c40fc commit d01c717

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

generation/maisi/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ When `autoencoder_sliding_window_infer_size` is equal to or larger than the late
8484

8585

8686
### Training GPU Memory Usage
87-
The VAE is trained on patches and can be trained using a 16G GPU if the patch size is set to a small value, such as [64, 64, 64]. Users can adjust the patch size to fit the available GPU memory. For the released model, we initially trained the autoencoder on a 16G V100 GPU with a small patch size of [64, 64, 64], and then continued training on a 32G V100 GPU with a larger patch size of [128, 128, 128].
87+
The VAE is trained on patches and can be trained using a 16G GPU if the patch size is set to a small value, such as [64, 64, 64]. Users can adjust the patch size to fit the available GPU memory. For the released model, we initially trained the autoencoder on 16G V100 GPUs with a small patch size of [64, 64, 64], and then continued training on 32G V100 GPUs with a larger patch size of [128, 128, 128].
8888

8989
The DM and ControlNet are trained on whole images rather than patches. The GPU memory usage during training depends on the size of the input images.
9090

0 commit comments

Comments
 (0)