[PyTorch Lightning] TensorBoard hp metric 제거

PyTorch Lightning 기반으로 프로젝트를 구현중이다.

 

Logger를 TensorBoard로 세팅하고 학습시키니 TensorBoard 파일에 hp metric이란 그래프가 떡하니 출력된다. 관련 글들을 좀 찾다보니 하이퍼 파라미터 서치에 도움을 주기 위한 그래프 같은데 굳이 출력할 필요가 없었다. 불필요한 정보가 출력되고 있으면 괜히 뭔가 마음이 불편하다. 근데 PyTorch Lightning이 캡슐화가 너무 잘 돼있다보니 어딜 건드려야 이 그래프를 지울 수 있는지 모르겠다... 구글링을 해보니 이미 나와 같은 생각을 하신 분이 계셨다.

 

 

 

해결법은 간단했다.

 

TensorBoardLogger 클래스의 객체를 생성할때 함수인자 default_parameter 를 False로 세팅해주면 된다.

 

    logger = TensorBoardLogger(...,
                               default_hp_metric=False)

    trainer = pl.Trainer(...,
                         logger=logger)

 

 

 

https://stackoverflow.com/questions/65450707/what-is-hp-metric-in-tensorboard-and-how-to-get-rid-of-it

What is hp_metric in TensorBoard and how to get rid of it?