[StarGAN] 코드 파악하기 - logger.py

 

 

기존 Logger

StarGAN version1이 2018년에 나오고 지금은 시간이 꽤 지나 version2까지 나온 상황이다. 그 사이에 tensorflow가 기존 1 버전의 문법과는 다른 부분이 많은 2 버전이 출시되었다.

tensorflow 2 버전 사용법에 관련해서는 www.tensorflow.org/guide/effective_tf2?hl=ko를 참조하면 도움이 될 것이다.

 

tensorflow 버전 확인

 

2 버전 사용 시 오류가 발생하는 곳이 logger.py이다. (1 버전을 사용하고 있다면 에러가 발생하지 않을 것이다)

tf.summary.FileWriter() 함수는 tensorflow 1 버전에서만 사용 가능한 함수다. 따라서 2 버전에서 지원하는 tf.summary.create_file_writer() 함수로 수정해줘야 한다. 

 

앞에 나온 링크에 따르면 Summary 로그를 기록하기 위한 방법은 위와 같다.

그래서 scalar_summary() 함수 내에서도 위와 같은 형식으로 수정해줘야 한다.

버전 2 사용 시에 최종 수정된 Logger 클래스는 다음과 같다.

 

수정 Logger