Skip to content

Commit

Permalink
May be fix for the issue in the visdom as in
Browse files Browse the repository at this point in the history
  • Loading branch information
Soumya Tripathy committed May 25, 2021
1 parent 4bf360f commit 13d3c87
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions util/visualizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,8 +100,9 @@ def display_current_results(self, visuals, epoch, save_result):
def plot_current_errors(self, epoch, counter_ratio, opt, errors):
if not hasattr(self, 'plot_data'):
self.plot_data = {'X': [], 'Y': [], 'legend': list(errors.keys())}
self.plot_data['X'].append(epoch + counter_ratio)
self.plot_data['Y'].append([errors[k] for k in self.plot_data['legend']])
self.plot_data['X'].append(float(epoch + counter_ratio))
self.plot_data['Y'].append([float(errors[k]) for k in self.plot_data['legend']])
import pdb; pdb.set_trace()
self.vis.line(
X=np.stack([np.array(self.plot_data['X'])] * len(self.plot_data['legend']), 1),
Y=np.array(self.plot_data['Y']),
Expand Down

0 comments on commit 13d3c87

Please sign in to comment.