MamaPearl
/

nula-cifar10-robust-v0

Image Classification

computer-vision

adversarial-robustness

Model card Files Files and versions

MamaPearl commited on Apr 1

Commit

7f87b44

·

verified ·

1 Parent(s): aa667e0

Update train_robust.py

Files changed (1) hide show

train_robust.py +1 -1

train_robust.py CHANGED Viewed

@@ -110,7 +110,7 @@ for epoch in range(1, NUM_EPOCHS + 1):
         print(f"new best: {100 * best_val_acc:.2f}%")
     current_lr = optimizer.param_groups[0]["lr"]
-    print("=", "-"*60, "=")
     print(f"Epoch [{epoch}/{NUM_EPOCHS}]")
     print(f"lr : {current_lr:.6f}")
     print(f"train_loss : {train_loss:.4f}")

         print(f"new best: {100 * best_val_acc:.2f}%")
     current_lr = optimizer.param_groups[0]["lr"]
+    print(f"{'|'}{'-'*60}{'|'}")
     print(f"Epoch [{epoch}/{NUM_EPOCHS}]")
     print(f"lr : {current_lr:.6f}")
     print(f"train_loss : {train_loss:.4f}")