Closed raki-1203 closed 2 years ago
모듈화작업 완료
train.py 부터 inference.py 모두 잘 돌아가는 것 확인했고
trainer_qa.py 에서 self.compute_metrics = None 해주는 부분과는 별로 상관이 없었음
물론 저걸 주석처리하면 돌아가기는 하지만 저게 원인이 아니었음
training_args.evaluation_strategy 얘한테 'steps' 라고 주니까 evaluate() 를 돌지않았던 거였음
아무튼 삽질끝에 작업 끝났고 한번 보시고 train.py 와 inference.py 는 간결해서 깔끔 👍
대신 utils_qa.py 는 길어진 건 함정
TrainingArguments 를 상속받아서 output_dir 를 고정해두고 사용할 수 있게 만들어놨구요
제 생각은 output_dir 은 건드리지 않고 project_name 을 바꿔가면서 사용하시는게 관리하기 좋지 않을까 싶어서
추가해놨습니다.
기본적인 arguments 들 설정해놨어요
커밋 메세지부터 내용정리까지 가독성있고 간단한 PR이네요. 고생하셨어요
모듈화작업 완료
train.py 부터 inference.py 모두 잘 돌아가는 것 확인했고
trainer_qa.py 에서 self.compute_metrics = None 해주는 부분과는 별로 상관이 없었음
물론 저걸 주석처리하면 돌아가기는 하지만 저게 원인이 아니었음
training_args.evaluation_strategy 얘한테 'steps' 라고 주니까 evaluate() 를 돌지않았던 거였음
아무튼 삽질끝에 작업 끝났고 한번 보시고 train.py 와 inference.py 는 간결해서 깔끔 👍
대신 utils_qa.py 는 길어진 건 함정
TrainingArguments 를 상속받아서 output_dir 를 고정해두고 사용할 수 있게 만들어놨구요
제 생각은 output_dir 은 건드리지 않고 project_name 을 바꿔가면서 사용하시는게 관리하기 좋지 않을까 싶어서
추가해놨습니다.
기본적인 arguments 들 설정해놨어요