Project-MONAI / tutorials

MONAI Tutorials
https://monai.io/started.html
Apache License 2.0
1.74k stars 662 forks source link

Maisi VAE notebook #1757

Closed Can-Zhao closed 1 month ago

Can-Zhao commented 1 month ago

Fixes # .

Description

Add MAISI notebook and adjust it to a setting that can run on 16G V100.

Checks

review-notebook-app[bot] commented 1 month ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

mingxin-zheng commented 1 month ago

Hi @Can-Zhao , the CI indicates the something may be wrong in the last two cells. Can you try running it locally?

Update: I finish running the notebook locally. The last cell fails probably due to out-of-memory in github CI V100 16G. Last mem use I capture during I run the notebook 2024/07/12 15:43:44.493, 16640 MiB

mingxin-zheng commented 1 month ago

I think the PR overall looks good to me. It would be better to keep the outputs of the last 3 cells of the notebook.

Can-Zhao commented 1 month ago

I think the PR overall looks good to me. It would be better to keep the outputs of the last 3 cells of the notebook.

May I ask if I need to finish the whole training process to generate the output cell?

KumoLiu commented 1 month ago

I think the PR overall looks good to me. It would be better to keep the outputs of the last 3 cells of the notebook.

May I ask if I need to finish the whole training process to generate the output cell?

Yes, and after that you can reset the epoch to a small number such as 2 then the ci will not run the whole training.

Can-Zhao commented 1 month ago

I think the PR overall looks good to me. It would be better to keep the outputs of the last 3 cells of the notebook.

May I ask if I need to finish the whole training process to generate the output cell?

Yes, and after that you can reset the epoch to a small number such as 2 then the ci will not run the whole training.

That takes a lot of time, I do not have another week...I did not use these datasets for released model anyway...

Can-Zhao commented 1 month ago

Hi @Can-Zhao , the CI indicates the something may be wrong in the last two cells. Can you try running it locally?

Update: I finish running the notebook locally. The last cell fails probably due to out-of-memory in github CI V100 16G. Last mem use I capture during I run the notebook 2024/07/12 15:43:44.493, 16640 MiB

I changed configs and it is resolved

Can-Zhao commented 1 month ago

I think the PR overall looks good to me. It would be better to keep the outputs of the last 3 cells of the notebook.

Added output cells for 2 epochs