torch.jit.frontend.UnsupportedNodeError: break statements aren't supported about mvae

[oskopek]

Hi Maciej!

Thanks for using the code. I fixed a version in the Makefile and uploaded a new version (see commit 96d851c). Please recreate your whole environment setup by first running:

conda deactivate pt
conda env remove -n pt

and then running all the README steps again. Thanks!

Your provided example should then work just fine:

(pt)  $ python -m mt.examples.run --dataset="mnist" --model="h2,s2,e2" --fixed_curvature=False

CUDA is not available.
Running on: cpu
#####
VAE Model: e2,h2,s2; Epochs: 500; Time: 2020-03-16T23:28:56.472804; Fixed curvature: False; Dataset: mnist
#####
	TrainEpoch 0:	{'bce': 147.69038701468037, 'kl': 9.85506391545199, 'elbo': -157.54545093013218, 'll': 0.0, 'mi': 0.0, 'cov_norm': 0.0, 'beta': 1.0, 'comp_000_h2/curvature': -0.008264462809917356, 'comp_001_s2/curvature': 0.008264462809917356, 'comp_002_e2/curvature': 0.0}
	TrainEpoch 1:	{'bce': 122.45547009470756, 'kl': 11.968708590422095, 'elbo': -134.42417868512945, 'll': 0.0, 'mi': 0.0, 'cov_norm': 0.0, 'beta': 1.0, 'comp_000_h2/curvature': -0.01, 'comp_001_s2/curvature': 0.01, 'comp_002_e2/curvature': 0.0}
	TrainEpoch 2:	{'bce': 117.72101712977751, 'kl': 12.220480436734427, 'elbo': -129.94149756651183, 'll': 0.0, 'mi': 0.0, 'cov_norm': 0.0, 'beta': 1.0, 'comp_000_h2/curvature': -0.012345679012345678, 'comp_001_s2/curvature': 0.012345679012345678, 'comp_002_e2/curvature': 0.0}

Let me know if it helps or not. In case it doesn't, can you also try running make tests and paste the output here? Thanks!

[macio232]

Now everything works. Thank you for fast reaction!

[oskopek]

Welcome!