1. 메인 잡고 거기서 arg따라 train val test 나눔
2. train val test 스크립트 따로 만들고 각각 실행