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()