[Plant_2021 대회] Pytorch기반 기본 베이스라인 성능 공유 [0.271]

[ghost]

백본으로 resnet34이용

BATCH = 16
EPOCHS = 5
LR = 0.0001
transforms.Resize((128, 128))
model = torchvision.models.resnet34()
model.fc = nn.Linear(512, NUM_CL, bias=True)
model = model.to(DEVICE)
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=LR)

자세히보려면 : https://www.kaggle.com/yoonsikjo/plant-2021-starter