antoine.carreaud67 commited on
Commit ·
68c5cf5
1
Parent(s): 190c89c
Sync train script formatting
Browse files- train/train.py +1 -1
train/train.py
CHANGED
|
@@ -504,7 +504,7 @@ def main(cfg_path: str = "configs/FlairHub.yaml"):
|
|
| 504 |
"train_loss": train_loss,
|
| 505 |
"val_loss": val_loss,
|
| 506 |
"val_mIoU": current_miou,
|
| 507 |
-
"val_mF1": val_metrics["mF1"],
|
| 508 |
}
|
| 509 |
print(f"LR={current_lr:.6f} | train_loss={train_loss:.4f} | val_loss={val_loss:.4f} | mIoU={current_miou:.4f}")
|
| 510 |
if use_wandb:
|
|
|
|
| 504 |
"train_loss": train_loss,
|
| 505 |
"val_loss": val_loss,
|
| 506 |
"val_mIoU": current_miou,
|
| 507 |
+
"val_mF1": val_metrics["mF1"],
|
| 508 |
}
|
| 509 |
print(f"LR={current_lr:.6f} | train_loss={train_loss:.4f} | val_loss={val_loss:.4f} | mIoU={current_miou:.4f}")
|
| 510 |
if use_wandb:
|