Michael Diskin hace 4 años
padre
commit
56c2e96173
Se han modificado 2 ficheros con 2 adiciones y 1 borrados
  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()