search

boostcampaitech2 / object-detection-level2-cv-03

object-detection-level2-cv-03 created by GitHub Classroom
2 stars 8 forks source link

mm detection에서 wandb epoch 설정하는 방법입니다. #11

Open larcane97 opened 2 years ago

larcane97 commented 2 years ago

/opt/conda/envs/detection/lib/python3.7/site-packages/mmcv/runner/hooks/logger/wandb.py에서 아래 코드의 주석 부분을 추가해주시면 됩니다!

 def log(self, runner):
        tags = self.get_loggable_tags(runner)
        if tags:
            tags['epoch'] = self.get_epoch(runner) # 추가된 코드
            if self.with_step:
                self.wandb.log(
                    tags, step=self.get_iter(runner), commit=self.commit)
            else:
                tags['global_step'] = self.get_iter(runner)
                self.wandb.log(tags, commit=self.commit)