From 7f07f556943fc682d3e178201077f85a854ef37a Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Thu, 14 Nov 2019 12:52:15 -0700
Subject: [PATCH] update finetune

---
 generator/simple/finetune.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/generator/simple/finetune.py b/generator/simple/finetune.py
index 9f5e917..ad5cdcc 100644
--- a/generator/simple/finetune.py
+++ b/generator/simple/finetune.py
@@ -13,6 +13,9 @@ file_name = "merged-first-person.txt"
 sess = gpt2.start_tf_sess()
 gpt2.finetune(sess,
               file_name,
+              multi_gpu=True,
+              batch_size=16,
+              sample_length=256,
               model_name=model_name,
               steps=10000)