argparse 2

[pytest] 테스트 코드에서 명령행 인자 전달하는 방법 (feat. argparse)

아래와 같이 argparse 모듈을 통해 명령행 인자를 입력받는 함수 main()이 있다. # main.py import argparse def main(): parser = argparse.ArgumentParser() parser.add_argument("--name", default="sangwon") args = parser.parse_args() return args.name if __name__ == "__main__": print(main()) main.py는 아래과 같이 실행할 수 있다. >> python main.py --name YoonSangWon YoonSangWon 그렇다면 테스트 코드에서 main 함수에 명령행 인자를 전달하기 위해서는 어떻게 해야 할까? monkeypatch f..

Backend/pytest 2023.06.15

argparse, omegaconf로 실험 환경 구성하기

ML 프로젝트에서 다양한 실험을 진행하다 보면 다양한 하이퍼 파라미터들과 여러 설정값을 수정해야 하는 일이 자주 발생한다. 그럴 때마다 코드를 직접 수정하게 되면 수정해야 할 부분을 찾아야 하는 일도 번거롭고, 제대로 값을 수정하지 못하고 실험이 진행되는 일이 빈번하다. 따라서 이번 포스팅에서는 configuration을 적절하게 관리하는 방법에 대해서 알아보자. argparse argparse를 이용하면 CLI에서 argument의 값을 입력받을 수 있다. CLI에서 입력받은 argument 값을 사용하기 위해서는 코드를 다음과 같이 구성해야 한다. parser 객체 생성 입력받을 argument 선언 argument 파싱 아래는 예시 코드이다. import argparse def get_args():..

AI/기타 2023.05.12