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()