updated dependencies and configs

environment.yml

@@ -1,23 +1,27 @@
 name: dancer-net
 channels:
-  - anaconda
+  - pytorch
+  - nvidia
   - conda-forge
+  - anaconda
 dependencies:
-  - torchvision
+  - python=3.10
   - pytorch
+  - torchaudio
+  - torchvision
   - numpy
   - pandas
   - seaborn
-  - python=3.10
   - matplotlib
-  - torchaudio
   - bs4
   - requests
   - bidict
   - tqdm
   - pytorch-lightning
   - rich
-  - pip
-    - gradio
-    - wakepy
-prefix: /opt/homebrew/Caskroom/miniforge/base/envs/dancer-net
+  - scikit-learn
+  - tensorboard
+  - pip:
+      - git+https://github.com/huggingface/transformers.git
+      - evaluate
+      - wakepy

models/config/train.yaml

@@ -39,7 +39,6 @@ trainer:
   max_epochs: 50
   min_epochs: 5
   fast_dev_run: False
-  track_grad_norm: 2
   # gradient_clip_val: 0.5
   overfit_batches: 1
 training_environment: