Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
jspark418 authored Apr 15, 2023
1 parent 08c99d2 commit d5049bc
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,8 @@
```
sudo docker pull gcr.io/kaggle-gpu-images/python:v121
```
* baseline_transformer.py를 실행하면 모델을 실행할 수 있습니다.

## Models
* 현재 feature extractor에 사용된 모델은 transformer 입니다.
* transformer 외에 다른 모델을 사용하고 싶으면 model.py 에 작성된 모델의 코드를 수정하시면 됩니다.
* model.py 는 transformer 모델이 들어있는데 코드만 수정하면 다른 모델을 사용할 수 있습니다.

0 comments on commit d5049bc

Please sign in to comment.