Skip to content

Latest commit

 

History

History
39 lines (37 loc) · 690 Bytes

회의.md

File metadata and controls

39 lines (37 loc) · 690 Bytes

20220627

  • CONFIG 항목 결정해오기
CONFIG = dict(
    S=7,
    B=2,
    C=20,
    seed=42,
    batch_size=4,
    epoch = 150,
    height=384,   # y
    width=384,    # x
    lambda_coord=5.,
    lambda_noobj=0.5,
    lr=1e-5,
    start_epoch=1,
    device=torch.device("mps" if torch.has_mps else "cpu"),
)

더 추가할만한 것들

  • backbone model
    • ex) vit_base_patch32_384
  • 학습관련
    • optimizer ex) Adam, AdamW
    • scheduler
    • earyly_stopping
  • Image Augmentation
    • transform
  • Fine Tuning Option
    • classifer X epoch
    • full model X epoch
  • Dataset / Dataloader options
    • root_path
    • download
    • shuffle
    • train/val
  • WanDB