Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feature#8 #9

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

feature#8 #9

wants to merge 1 commit into from

Conversation

seungheondoh
Copy link
Collaborator

Pull Request

add src/task runner and pipeline
change train.py, evaluation.py with pytorch lightning

해당 PR을 제출하기 전에 아래 사항이 완료되었는지 확인 부탁드립니다:

  • 작성한 코드가 어떤 에러나 경고없이 빌드가 되었나요?
  • 충분한 테스트를 수행하셨나요?

1. 해당 PR은 어떤 내용인가요?

add src/task runner and pipeline

  • runner와 pipiline을 pytorchlightning를 활용하여 구성하였으며,
  • change train.py, evaluation.py 를 위 pytorchlightning 모듈을 통해 학습시켰습니다.
  • hparmas.py를 제거하고 conf 폴더 내의 yaml 파일로 통일시켰습니다.

2. PR과 관련된 이슈가 있나요?

  • Valid Epoch end의 경우 Sanity Check에 의해서 하나의 Batch로 Test하는 경우가 발생합니다. 하지만 Multi-label Classification이나 Autotagging같은 Task의 경우 ROCAUC를 Metric으로 사용하는데 이때, 전체 레이블마다 적어도 하나의 binary 데이터가 필요합니다. Batch 사이즈가 작으면 이 경우 특정 imbalance한 label이 없어서 에러가 발생하는 경우가 있습니다. metric.py에 간단한 에러처리를 해두었습니다.

add src/task runner and pipeline
change train.py, evaluation.py with pytorchlightning
Comment on lines +61 to +71
if results_path.exists():
with open(results_path, mode="r") as io:
results = json.load(io)

result = trainer.test(runner, test_dataloaders=dataloader)
results.update({"checkpoint": args.checkpoint, f"{args.type}": result})

else:
results = {}
result = trainer.test(runner, test_dataloaders=dataloader)
results.update({"checkpoint": args.checkpoint, f"{args.type}": result})
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

이 구문의 의도가 무엇인지 알 수 있을까요?

Copy link
Member

@ssaru ssaru left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@Dohppak 코멘트 완료하였습니다.
*.ckpt 파일이 올라와있던데, 일부로 올리신걸까요?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants