Michael Diskin há 4 anos atrás
pai
commit
56c2e96173
2 ficheiros alterados com 2 adições e 1 exclusões
  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()