From d5049bc34ad80a6c1d106f3cccfde11642cb1eee Mon Sep 17 00:00:00 2001 From: Park Jongseong <92840498+jspark418@users.noreply.github.com> Date: Sat, 15 Apr 2023 16:24:32 +0900 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index fa22346..b8cd537 100644 --- a/README.md +++ b/README.md @@ -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 모델이 들어있는데 코드만 수정하면 다른 모델을 사용할 수 있습니다.