gradboard 2.3.1__tar.gz → 2.3.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of gradboard might be problematic. Click here for more details.

@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: gradboard
3
- Version: 2.3.1
3
+ Version: 2.3.3
4
4
  Summary: Easily snowboard down gnarly loss gradients
5
5
  License: MIT
6
6
  Author: Nicholas Bailey
@@ -136,13 +136,14 @@ def get_optimiser(model, optimiser=AdamW, lr=1e-3, weight_decay=1e-2):
136
136
  https://github.com/SHI-Labs/Compact-Transformers/blob/main/configs/
137
137
  pretrained/cct_7-3x1_cifar100_1500epochs.yml
138
138
  """
139
+ # TODO: print a warning when params are excluded from weight decay IFF wd is set
139
140
  weight_decay_exclude = []
140
141
  for keyword in [
142
+ "nondecay",
141
143
  "bias",
142
144
  "norm",
143
145
  "embedding",
144
- "swiglu_beta",
145
- "sigma_reparam",
146
+ "beta",
146
147
  "input_query",
147
148
  "reentrant_query",
148
149
  ]:
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "gradboard"
3
- version = "2.3.1"
3
+ version = "2.3.3"
4
4
  description = "Easily snowboard down gnarly loss gradients"
5
5
  authors = [
6
6
  {name = "Nicholas Bailey"}
File without changes
File without changes
File without changes