@@ -244,7 +244,7 @@ class Main(object):
init_model_path,
mode=int(model_mode))
epoch_start_time = time.time()
-
+ epoch = 0
if sync_mode == "heter":
self.heter_train_loop(epoch)
elif reader_type == "QueueDataset":