How to load MagentaJS musicVAE model into Python magenta MusicVAE

89 Views Asked by At

I am attempting to load the MusicVAE model from the Drumbot Project into a python MusicVAE model.

However, the constructors for each are not the same. The Javascript version accepts a URL to a folder(link here) which contains 3 files: a weights_manifest.json, a shard of data, and a config.json file.

The python implementation takes 2 mandatory arguments: a config object, and a checkpoint directory containing checkpoint files.

These two seem to be incompatible, and I'm not sure how I would go about converting the JS checkpoint directory into something usable with the Python model. I've looked into the tfjs_converter utility, but it does not accept the format that the JS version is in as an input.

0

There are 0 best solutions below