Skip to content

Commit b98e030

Browse files
authored
fix: correct CSV logger column headers to match logged data (#43)
- Change 'loss' column to 'train_acc' - Change 'acc' column to 'val_acc' - Headers now accurately reflect train_acc and val_acc values being logged
1 parent 2edefcb commit b98e030

File tree

1 file changed

+1
-1
lines changed
  • evals/video_classification_frozen

1 file changed

+1
-1
lines changed

evals/video_classification_frozen/eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def main(args_eval, resume_preempt=False):
133133

134134
# -- make csv_logger
135135
if rank == 0:
136-
csv_logger = CSVLogger(log_file, ("%d", "epoch"), ("%.5f", "loss"), ("%.5f", "acc"))
136+
csv_logger = CSVLogger(log_file, ("%d", "epoch"), ("%.5f", "train_acc"), ("%.5f", "val_acc"))
137137

138138
# Initialize model
139139

0 commit comments

Comments
 (0)