From 8402292704e89e41cc548dc94ae3c36a2f44cdda Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Tue, 5 Mar 2019 17:59:56 +0200
Subject: [PATCH] language model example: fix code formatting
---
examples/word_language_model/main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/word_language_model/main.py b/examples/word_language_model/main.py
index 1ad14db..6e1cb71 100755
--- a/examples/word_language_model/main.py
+++ b/examples/word_language_model/main.py
@@ -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)
--
GitLab