def __init__(self, vis_screen):
self.viz = VisdomPlotter(env_name=vis_screen)
self.hist_D = []
self.hist_G = []
self.hist_E = []
self.hist_Dx = []
self.hist_DGx = []
def __init__(self, vis_screen, save_path):
self.viz = VisdomPlotter(env_name=vis_screen)
self.hist_D = []
self.hist_G = []
self.hist_Dx = []
self.hist_DGx = []
self.logger = logging.getLogger('lossesLogger')
self.logFile = os.path.join('logs', save_path)
if not os.path.exists(self.logFile):
os.makedirs(self.logFile)
handler = logging.FileHandler(self.logFile + '/logFile.log')
handler.setLevel(logging.INFO)
self.logger.addHandler(hdlr=handler)
self.logger.setLevel(logging.INFO)