a901f23f1d47519508df7d96eef815c725ef3d3a,transcribe.py,,,#,98

Before Change

    out = model(Variable(spect, volatile=True))
    out = out.transpose(0, 1)  // TxNxH

    if args.word_json:
        results = word_decode(decoder, out.data, model=model, window=1 / (10 * audio_conf["window_size"]),
                              time_div=1 / audio_conf["window_size"])
        print(json.dumps(results))
    else:
        decoded_output, decoded_offsets = decoder.decode(out.data)
        for b in range(len(decoded_output)):
            for pi in range(max(args.top_paths, len(decoded_output[b]))):
                print(decoded_output[b][pi])
                if args.offsets:
                    print(decoded_offsets[b][pi])

After Change

    out = model(Variable(spect, volatile=True))
    out = out.transpose(0, 1)  // TxNxH
    decoded_output, decoded_offsets = decoder.decode(out.data)
    print(json.dumps(decode_results(decoded_output, decoded_offsets)))

Instances