Skip to content

Commit c872e4b

Browse files
committed
Minor fix to plotting script.
1 parent 1ef29d1 commit c872e4b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/plot_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def main():
146146
parser.add_argument('--labels', nargs='+', default=[],
147147
help='Labels for each model used in the plot. There should be as many '
148148
'labels as models.')
149-
parser.add_argument('--metric', default='perplexity-diff',
149+
parser.add_argument('--metric', default='cross-entropy-diff',
150150
help='Which metric to plot on the y-axis. Default is difference in '
151151
'cross entropy between the model and the lower bound on the '
152152
'validation set.')

0 commit comments

Comments
 (0)