diff --git a/yolov5/utils/loggers/neptune/neptune_utils.py b/yolov5/utils/loggers/neptune/neptune_utils.py index ff12486..64537fa 100644 --- a/yolov5/utils/loggers/neptune/neptune_utils.py +++ b/yolov5/utils/loggers/neptune/neptune_utils.py @@ -15,7 +15,7 @@ def __init__(self, opt, job_type='Training'): self.job_type = job_type with open(opt.data) as f: data_dict = yaml.safe_load(f) # data dict - self.neptune, self.neptune_run = neptune, None, data_dict + self.neptune, self.neptune_run = neptune, None if self.neptune and opt.neptune_token: self.neptune_run = neptune.init(api_token=opt.neptune_token,