Michael Diskin 4 anni fa
parent
commit
56c2e96173
2 ha cambiato i file con 2 aggiunte e 1 eliminazioni
  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()