3D-VQ-VAE-2

3D-VQ-VAE-2 copied to clipboard

27yw

Steps for Training VQ-VAE?

2

Hello, I wanted to confirm the steps for training a VQ-VAE on radiology data. Thank you for working on such an interesting and important application of VQ-VAE. Our research group...

aksg87

Clarifications on Decoder and Embeddings

4

@robogast Happy to report I was able to train a VQ-VAE using a dataset. Very cool to see - and kudos for the nice Tensorboard outputs you have in place!...

aksg87

Installing

2

When installing the dependencies, I am getting: ``` $ conda env create -f environment.yml Collecting package metadata (repodata.json): done Solving environment: failed ResolvePackageNotFound: - torchvision==0.9.0.dev20210122=py38_cu110 - pytorch==1.8.0.dev20210122=py3.8_cuda11.0.221_cudnn8.0.5_0 ``` Any ideas?

aksg87