Questions about new classes

[1144181135]

Hi jukebox teams, I have some questions about training and sampling on new categories. I have trained a prior (level 2) on my defined 18 categories, and I uses y_bins=(18, 1) (the original is 120,4111) in "labels_v3" and "small_labelled_prior". I try to use my trained prior to sample music as ReadMe with your pretrained vqvae, upsample_level0 and upsample_level1. But when loading upsample_level0 and upsample_level1, loading error happens. It seems that my defined y_bins is (18, 1) but your defined y_bins is (120, 4111), so the weight dimension "self.y_emb = LabelConditioner" is different in your pretrained upsampler. In such a situation, what should I do? Using a random "self.y_emb" weight ? Or retraining the upsampler on my defined 18 categories with y_bins = (18, 1) ? Or Any ? Please give me some advice. Thank you!