joenathan commited on
Commit
95cdfc0
·
verified ·
1 Parent(s): 476b2b3

Update docstrings for the renamed notebook

Browse files
Files changed (2) hide show
  1. model.py +1 -1
  2. train.py +1 -1
model.py CHANGED
@@ -1,5 +1,5 @@
1
  """Shared model/Grad-CAM code, extracted from
2
- Problem10_Explainable_Auditable_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
 
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
- (Problem10_Explainable_Auditable_Defect_Detection.ipynb) -- this script exists only
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