glenn-jocher commited on
Commit
9dbaa1b
·
1 Parent(s): 83dc540

Update plot_results()

Browse files
Files changed (1) hide show
  1. utils/general.py +2 -2
utils/general.py CHANGED
@@ -143,7 +143,7 @@ def check_dataset(dict):
143
  if not all(os.path.exists(x) for x in val):
144
  print('\nWARNING: Dataset not found, nonexistant paths: %s' % [*val])
145
  if s and len(s): # download script
146
- print('Attempting autodownload from: %s' % s)
147
  if s.startswith('http') and s.endswith('.zip'): # URL
148
  f = Path(s).name # filename
149
  if platform.system() == 'Darwin': # avoid MacOS python requests certificate error
@@ -1263,7 +1263,7 @@ def plot_results(start=0, stop=0, bucket='', id=(), labels=(),
1263
  y[y == 0] = np.nan # dont show zero loss values
1264
  # y /= y[0] # normalize
1265
  label = labels[fi] if len(labels) else Path(f).stem
1266
- ax[i].plot(x, y, marker='.', label=label, linewidth=2, markersize=8)
1267
  ax[i].set_title(s[i])
1268
  # if i in [5, 6, 7]: # share train and val loss y axes
1269
  # ax[i].get_shared_y_axes().join(ax[i], ax[i - 5])
 
143
  if not all(os.path.exists(x) for x in val):
144
  print('\nWARNING: Dataset not found, nonexistant paths: %s' % [*val])
145
  if s and len(s): # download script
146
+ print('Downloading %s ...' % s)
147
  if s.startswith('http') and s.endswith('.zip'): # URL
148
  f = Path(s).name # filename
149
  if platform.system() == 'Darwin': # avoid MacOS python requests certificate error
 
1263
  y[y == 0] = np.nan # dont show zero loss values
1264
  # y /= y[0] # normalize
1265
  label = labels[fi] if len(labels) else Path(f).stem
1266
+ ax[i].plot(x, y, marker='.', label=label, linewidth=1, markersize=6)
1267
  ax[i].set_title(s[i])
1268
  # if i in [5, 6, 7]: # share train and val loss y axes
1269
  # ax[i].get_shared_y_axes().join(ax[i], ax[i - 5])