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

fix: typo

Browse source 
Former-commit-id: 837b298afc0a8283bd18d0d4353c8f0491113b9a [formerly d64ed0d75f62f99b0be0a919d6d6bfffb6e5cd6f]
Former-commit-id: 964fb5b2dde3e90265dba16aef747b34dcfb6fb6
This commit is contained in:
Laurent Fainsin 2022-06-29 10:26:26 +02:00
parent de7e2e895f
commit 83f8825340
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/train.py
View file

@ -17,8 +17,8 @@ from unet import UNet
from utils.paste import RandomPaste from utils.paste import RandomPaste
CHECKPOINT_DIR = Path("./checkpoints/") CHECKPOINT_DIR = Path("./checkpoints/")
DIR_TRAIN_IMG = Path("/home/lilian/data_disk/lfainsin/smoltrain2017") DIR_TRAIN_IMG = Path("/home/lilian/data_disk/lfainsin/val2017")
DIR_VALID_IMG = Path("/home/lilian/data_disk/lfainsin/smolval2017/") DIR_VALID_IMG = Path("/home/lilian/data_disk/lfainsin/val2017/")
DIR_SPHERE_IMG = Path("/home/lilian/data_disk/lfainsin/spheres/Images/") DIR_SPHERE_IMG = Path("/home/lilian/data_disk/lfainsin/spheres/Images/")
DIR_SPHERE_MASK = Path("/home/lilian/data_disk/lfainsin/spheres/Masks/") DIR_SPHERE_MASK = Path("/home/lilian/data_disk/lfainsin/spheres/Masks/")
@ -182,7 +182,7 @@ def main():
# forward # forward
with torch.cuda.amp.autocast(enabled=args.amp): with torch.cuda.amp.autocast(enabled=args.amp):
pred_masks = net(images) pred_masks = net(images)
train_loss = criterion(pred_masks, pred_masks) train_loss = criterion(pred_masks, true_masks)
# backward # backward
optimizer.zero_grad(set_to_none=True) optimizer.zero_grad(set_to_none=True)