Step 2.3 - Train Dense Teacher

Signed-off-by: Francesco Murdaca <[email protected]>
os-climate · Mar 16, 2022 · df0e182 · df0e182
1 parent ef29347
commit df0e182
Show file tree

Hide file tree

Showing 3 changed files with 1,646 additions and 0 deletions.
diff --git a/notebooks/demo-nm/dense-teacher-training.ipynb b/notebooks/demo-nm/dense-teacher-training.ipynb
diff --git a/overlays/nm-dense-teacher/Pipfile b/overlays/nm-dense-teacher/Pipfile
@@ -0,0 +1,14 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[packages]
+sparseml = "*"
+torch = "<1.9"
+boto3 = "*"
+
+[dev-packages]
+
+[requires]
+python_version = "3.8"