Skip to content
Snippets Groups Projects
Commit 84022927 authored by Neta Zmora's avatar Neta Zmora
Browse files

language model example: fix code formatting

parent cee295fb
No related branches found
No related tags found
No related merge requests found
......@@ -271,7 +271,7 @@ def train(epoch, optimizer, compression_scheduler=None):
def export_onnx(path, batch_size, seq_len):
msglogger.info('The model is also exported in ONNX format at {}'.
format(os.path.realpath(args.onnx_export)))
format(os.path.realpath(args.onnx_export)))
model.eval()
dummy_input = torch.LongTensor(seq_len * batch_size).zero_().view(-1, batch_size).to(device)
hidden = model.init_hidden(batch_size)
......@@ -297,7 +297,7 @@ if args.summary:
bottomk, _ = torch.topk(param.abs().view(-1), int(percentile * param.numel()),
largest=False, sorted=True)
threshold = bottomk.data[-1]
msglogger.info("parameter %s: q = %.2f" %(name, threshold))
msglogger.info("parameter %s: q = %.2f" % (name, threshold))
else:
distiller.model_summary(model, which_summary, 'wikitext2')
exit(0)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment