백본으로 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
백본으로 resnet34이용
자세히보려면 : https://www.kaggle.com/yoonsikjo/plant-2021-starter