This commit is contained in:
William Falcon
2019-07-08 20:12:27 -04:00
parent 65f6cd4321
commit 12ee3c60dd
+4
View File
@@ -1,3 +1,6 @@
"""
The trainer handles all the logic for running a val loop, training loop, distributing, etc...
"""
from time import sleep
import subprocess
import traceback
@@ -23,6 +26,7 @@ try:
except ModuleNotFoundError:
APEX_AVAILABLE = False
class Trainer(TrainerIO):
def __init__(self,