Facing an error in running train_diffusion.py command. This is in the 2d_ldm folder, following the generative tutorial for 2D

[programmerwali]

When I try to run the code of 2d_ldm in generative in the tutorial repository. I followed the read me and reached to the step where I have to run this code. This is to run the train_diffusion.py file:

!python train_diffusion.py -c ./config/config_train_32g.json -e ./config/environment.json -g 1

This is the error that I get:

FileNotFoundError: [Errno 2] No such file or directory: './trained_weights/diffusion_2d/autoencoder.pt'

How to resolve the auto encoder.pt error?

I have my environment set up on Google Colab and using its GPU.

[programmerwali]

[KumoLiu]

Hi @programmerwali, we didn't provide the pre-trained weight in the tutorial repo. You can find the weights in the MONAI Zoo repo.
https://github.com/Project-MONAI/model-zoo/blob/dev/models/brats_mri_axial_slices_generative_diffusion/large_files.yml
https://github.com/Project-MONAI/model-zoo/blob/dev/models/brats_mri_generative_diffusion/large_files.yml

Hope it helps, thanks!