Skip to content

Initial DiffusionWaveGAN pretrained

Latest
Compare
Choose a tag to compare
@revsic revsic released this 19 Mar 07:52
from config import Config
from dwg import DiffusionWaveGAN

with open('vlb-stft-unweighted.json') as f:
    config = Config.load(json.load(f))

ckpt = torch.load('vlb-stft-unweighted_460.ckpt', map_location='cpu')

dwg = DiffusionWaveGAN(config.model)
dwg.load(ckpt)
dwg.eval()