Update train/train_seg.py
Browse files- train/train_seg.py +1 -2
train/train_seg.py
CHANGED
|
@@ -97,8 +97,7 @@ def train(args):
|
|
| 97 |
|
| 98 |
pbar = tqdm(train_loader, desc=f"Epoch {epoch}/{args.epochs}", leave=False)
|
| 99 |
for step, (images, masks) in enumerate(pbar):
|
| 100 |
-
|
| 101 |
-
break
|
| 102 |
images, masks = images.to(device), masks.to(device)
|
| 103 |
|
| 104 |
out = model(images)
|
|
|
|
| 97 |
|
| 98 |
pbar = tqdm(train_loader, desc=f"Epoch {epoch}/{args.epochs}", leave=False)
|
| 99 |
for step, (images, masks) in enumerate(pbar):
|
| 100 |
+
|
|
|
|
| 101 |
images, masks = images.to(device), masks.to(device)
|
| 102 |
|
| 103 |
out = model(images)
|