DeepLearning|MachineLearning/연구하며 사소한 실수들1 train, validation 과정에서 +=연산과 append, extend 딥러닝 학습과 검증 과정을 할 때def validation(model, criterion, val_loader, device): model.eval() val_loss = [] preds, true_labels = [], [] with torch.no_grad(): for imgs, labels in tqdm(iter(val_loader)): imgs = imgs.float().to(device) labels = labels.to(device) pred = model(imgs) loss = criterion(pred, labels) .. 2025. 4. 10. 이전 1 다음 반응형