Skip to content

Commit ad37a17

Browse files
committed
Fixed code formatting issue
1 parent 3e049c2 commit ad37a17

File tree

2 files changed

+0
-5
lines changed

2 files changed

+0
-5
lines changed

examples/references/classification/imagenet/dataflow.py

-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
from typing import Callable, Optional, Tuple
33

44
import cv2
5-
import numpy as np
65

76
import torch
87
from torch.utils.data import DataLoader

examples/references/classification/imagenet/main.py

-4
Original file line numberDiff line numberDiff line change
@@ -383,10 +383,6 @@ def evaluation(local_rank, config, logger, with_clearml):
383383
tb_logger = common.TensorboardLogger(log_dir=config.output_path.as_posix())
384384
tb_logger.attach_output_handler(evaluator, event_name=Events.COMPLETED, tag="validation", metric_names="all")
385385

386-
# Log confusion matrix to ClearML:
387-
if with_clearml:
388-
evaluator.add_event_handler(Events.COMPLETED, compute_and_log_cm, cm_metric, evaluator.state.iteration)
389-
390386
state = evaluator.run(data_loader)
391387
utils.log_metrics(logger, 0, state.times["COMPLETED"], "Validation", state.metrics)
392388

0 commit comments

Comments
 (0)