Miscellaneous fixes

konstantinjdobler · Jul 31, 2023 · 76d0995 · 76d0995
1 parent 06f90e5
commit 76d0995
Show file tree

Hide file tree

Showing 6 changed files with 1,640 additions and 1,563 deletions.
diff --git a/conda-lock.yml b/conda-lock.yml
@@ -9,7 +9,7 @@
 # To update a single package to the latest version compatible with the version constraints in the source:
 #     conda-lock lock  --lockfile conda-lock.yml --update PACKAGE
 # To re-solve the entire environment, e.g. after changing a version constraint in the source file:
-#     conda-lock -f ./environment.yml -f /Users/kjd/prog/nlp-research-template/environment.yml --lockfile conda-lock.yml
+#     conda-lock -f ./environment.yml --lockfile conda-lock.yml
 metadata:
   channels:
   - url: conda-forge
@@ -28,7 +28,6 @@ metadata:
   - win-64
   sources:
   - ./environment.yml
-  - /Users/kjd/prog/nlp-research-template/environment.yml
 package:
 - category: main
   dependencies: {}

diff --git a/dlib/frameworks/huggingface.py b/dlib/frameworks/huggingface.py
@@ -0,0 +1,7 @@
+import transformers
+
+
+class EvaluateFirstStepCallback(transformers.TrainerCallback):
+    def on_step_end(self, args, state, control, **kwargs):
+        if state.global_step == 1:
+            control.should_evaluate = True
diff --git a/dlib/frameworks/wandb.py b/dlib/frameworks/wandb.py
@@ -13,8 +13,8 @@
 
 from ..frameworks.pytorch import get_rank
 
-WANDB_PROJECT = "test-wandb"
-WANDB_ENTITY = "konstantinjdobler"
+WANDB_PROJECT = "<your project>"
+WANDB_ENTITY = "<your entity>"
 
 if WANDB_ENTITY == "<your entity>" or WANDB_PROJECT == "<your project>":
     logger.warning(