Update Mo.py

Mo.py

@@ -99,7 +99,8 @@ def txt_stream(file_path):
             )
 
 
-LIMIT = 36757266 // 2
+steps_per_epoch = 23119910 // batch_size
+LIMIT = 23119910
 
 dataset = tf.data.Dataset.from_generator(
     lambda: txt_stream(DATA_PATH),
@@ -264,7 +265,7 @@ with strategy.scope():
     model.compile(optimizer=optimizer, loss=smoothed_loss_keras, metrics=[masked_perplexity])
 
     # 학습
-    history = model.fit(dist_dataset, epochs=1, verbose=1)
+    history = model.fit(dist_dataset, epochs=1, steps_per_epoch=steps_per_epoch, verbose=1)
 
 model.save_weights("tf_model.weights.h5")
 print("✅ 모델 가중치 저장 완료!")