Ejemplos de WaveNODE.parameters en Python

Ejemplo n.º 1

def build_model(hps, log):
    model = WaveNODE(hps)
    n_params = sum(p.numel() for p in model.parameters() if p.requires_grad)
    print('number of parameters:', n_params)
    state = {}
    state['n_params'] = n_params
    log.write('%s\n' % json.dumps(state))
    log.flush()

    return model

Ejemplo n.º 2

Archivo: test_speed.py Proyecto: wavenode-anonymous/WaveNODE

def build_model(hps):
    model = WaveNODE(hps)
    print('number of parameters:',
          sum(p.numel() for p in model.parameters() if p.requires_grad))

    return model