Update docstrings for the renamed notebook
Browse files
model.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
"""Shared model/Grad-CAM code, extracted from
|
| 2 |
-
|
| 3 |
Space app and the training script can both import it without duplication."""
|
| 4 |
import numpy as np
|
| 5 |
import torch
|
|
|
|
| 1 |
"""Shared model/Grad-CAM code, extracted from
|
| 2 |
+
mit-group8-explainable-defect-detection.ipynb (sections 3 and 6) so the
|
| 3 |
Space app and the training script can both import it without duplication."""
|
| 4 |
import numpy as np
|
| 5 |
import torch
|
train.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
"""Trains the DefectCNN on the local NEU-DET mirror and saves weights for the Space app.
|
| 2 |
Reuses the same architecture/hyperparameters as the main notebook's section 3-4
|
| 3 |
-
(
|
| 4 |
because the notebook itself never persisted a checkpoint to disk."""
|
| 5 |
import glob
|
| 6 |
import os
|
|
|
|
| 1 |
"""Trains the DefectCNN on the local NEU-DET mirror and saves weights for the Space app.
|
| 2 |
Reuses the same architecture/hyperparameters as the main notebook's section 3-4
|
| 3 |
+
(mit-group8-explainable-defect-detection.ipynb) -- this script exists only
|
| 4 |
because the notebook itself never persisted a checkpoint to disk."""
|
| 5 |
import glob
|
| 6 |
import os
|