Michael Diskin 4 年 前
コミット
56c2e96173
2 ファイル変更2 行追加1 行削除
  1. 1 1
      examples/albert/TPU.py
  2. 1 0
      examples/albert/run_trainer.py

+ 1 - 1
examples/albert/TPU.py

@@ -79,4 +79,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

+ 1 - 0
examples/albert/run_trainer.py

@@ -312,5 +312,6 @@ def _mp_fn(index):
     # For xla_spawn (TPUs)
     main()
 
+
 if __name__ == "__main__":
     main()