SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix indentation weirdness in GPT-2 example.

Browse Source
This commit is contained in:
Sangwhan Moon
2019-04-22 02:20:22 +09:00
parent 68a889ee43
commit 14b1f719f4
1 changed files with 17 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/run_gpt2.py
View File

@@ -107,7 +107,7 @@ def run_model():
print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40) print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
print(text) print(text)
print("=" * 80) print("=" * 80)
if args.unconditional: else:
generated = 0 generated = 0
for _ in range(args.nsamples // args.batch_size): for _ in range(args.nsamples // args.batch_size):
out = sample_sequence( out = sample_sequence(
@@ -124,8 +124,6 @@ def run_model():
print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40) print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
print(text) print(text)
print("=" * 80) print("=" * 80)
if args.unconditional:
break
if __name__ == '__main__': if __name__ == '__main__':
run_model() run_model()