ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix: moved a line wrongly

Browse source 
Former-commit-id: 93f2c7ee9a1926248716328dc9dd5a46395373be [formerly 598da90e69422bb29bf0c2770b789773c6360981]
Former-commit-id: 354585b269180f4785f8bd05164eb8c8b0d16dba
This commit is contained in:
Laurent Fainsin 2022-07-04 15:38:49 +02:00
parent c69579f9da
commit 55c4d05d2a
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/train.py
View file

@ -174,6 +174,7 @@ if __name__ == "__main__":
# forward
with torch.cuda.amp.autocast(enabled=wandb.config.AMP):
pred_masks = net(images)
train_loss = criterion(pred_masks, true_masks)
# backward
optimizer.zero_grad(set_to_none=True)
@ -182,7 +183,6 @@ if __name__ == "__main__":
grad_scaler.update()
# compute metrics
train_loss = criterion(pred_masks, true_masks)
pred_masks_bin = (torch.sigmoid(pred_masks) > 0.5).float()
accuracy = (true_masks == pred_masks_bin).float().mean()
dice = dice_coeff(pred_masks_bin, true_masks)
@ -203,7 +203,7 @@ if __name__ == "__main__":
}
)
if step and (step % 100 == 0 or step == len(train_loader)):
if step and (step % 250 == 0 or step == len(train_loader)):
# Evaluation round
net.eval()
accuracy = 0