Skip to content

Commit

Permalink
grammar
Browse files Browse the repository at this point in the history
Signed-off-by: Can-Zhao <[email protected]>
  • Loading branch information
Can-Zhao committed Nov 21, 2024
1 parent f4c40fc commit d01c717
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion generation/maisi/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ When `autoencoder_sliding_window_infer_size` is equal to or larger than the late


### Training GPU Memory Usage
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].
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].

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.

Expand Down

0 comments on commit d01c717

Please sign in to comment.