- 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
- ex)
- 학습관련
optimizer
ex) Adam, AdamWscheduler
- earyly_stopping
- Image Augmentation
transform
- Fine Tuning Option
- classifer
X
epoch - full model
X
epoch
- classifer
- Dataset / Dataloader options
- root_path
- download
- shuffle
- train/val
- WanDB