Skip to content

Latest commit

 

History

History
15 lines (12 loc) · 713 Bytes

README.md

File metadata and controls

15 lines (12 loc) · 713 Bytes

outfittransformer

environment setting

  • pip install -r requirement.txt

dataset

  • Google Drive에서 download (original: AIHub 'KFashion 이미지' 데이터)
  • outfittransformer/data 폴더 생성
  • compatibility_train.txt, compatibility_valid.txt, item_metadata.json 파일 해당 경로로 이동

train

  • model_arg.py에서 주요 hyperparameter 수정 가능
python cp_train.py --task cp --train_batch 64 --valid_batch 96 --n_epochs 5 --learning_rate 1e-5 --scheduler_step_size 1000