From 83f882534084d17c492a35a94213a66b161cd8d6 Mon Sep 17 00:00:00 2001 From: Laurent Fainsin Date: Wed, 29 Jun 2022 10:26:26 +0200 Subject: [PATCH] fix: typo Former-commit-id: 837b298afc0a8283bd18d0d4353c8f0491113b9a [formerly d64ed0d75f62f99b0be0a919d6d6bfffb6e5cd6f] Former-commit-id: 964fb5b2dde3e90265dba16aef747b34dcfb6fb6 --- src/train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/train.py b/src/train.py index b286cfa..e9c63dc 100644 --- a/src/train.py +++ b/src/train.py @@ -17,8 +17,8 @@ from unet import UNet from utils.paste import RandomPaste CHECKPOINT_DIR = Path("./checkpoints/") -DIR_TRAIN_IMG = Path("/home/lilian/data_disk/lfainsin/smoltrain2017") -DIR_VALID_IMG = Path("/home/lilian/data_disk/lfainsin/smolval2017/") +DIR_TRAIN_IMG = Path("/home/lilian/data_disk/lfainsin/val2017") +DIR_VALID_IMG = Path("/home/lilian/data_disk/lfainsin/val2017/") DIR_SPHERE_IMG = Path("/home/lilian/data_disk/lfainsin/spheres/Images/") DIR_SPHERE_MASK = Path("/home/lilian/data_disk/lfainsin/spheres/Masks/") @@ -182,7 +182,7 @@ def main(): # forward with torch.cuda.amp.autocast(enabled=args.amp): pred_masks = net(images) - train_loss = criterion(pred_masks, pred_masks) + train_loss = criterion(pred_masks, true_masks) # backward optimizer.zero_grad(set_to_none=True)