Steps for Training VQ-VAE? about 3d-vq-vae-2 HOT 2 OPEN

Just an update!

I was able to get the train.py script running which is cool 🙂. Will share a few details in case anyone finds them helpful.

1. I created a folder with a bunch of .nrrd volume files which is the only input I am using right now for train.py

2. The code initially threw in an error as it expected a list for num_embeddings but the default is an int so I did the following.
   vqvae/model.py

        # assert len(args.num_embeddings) in (1, args.n_bottleneck_blocks)  # expects list     
        self.num_embeddings = [args.num_embeddings for _ in range(args.n_bottleneck_blocks)]

        # if len(args.num_embeddings) == 1:
        #     self.num_embeddings = [args.num_embeddings[0] for _ in range(args.n_bottleneck_blocks)]
        # else:
        #     self.num_embeddings = args.num_embeddings

3. I added a batch-size default of 1 as I otherwise received an error:
   vqvae/train.py

parser.add_argument("--batch-size", type=int, default=1)

from 3d-vq-vae-2.

thanks for your answer! It helps a lot. I am trying to training it as well and could you help me with my dataset_path, pls? I wanna train the data named like this :BraTS2021_00377_t1.nii.gz. Does it mean that I put all the files in one dir and what do I need to change the code and where?

from 3d-vq-vae-2.