Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
1b559c6
1
Parent(s):
9978534
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
@@ -165,6 +165,7 @@ class LightningModel(LightningModule):
|
|
165 |
self.tokenizer = tokenizer
|
166 |
self.output = output
|
167 |
self.learning_rate = learning_rate
|
|
|
168 |
self.weight_decay = weight_decay
|
169 |
|
170 |
def forward(self, input_ids, attention_mask, decoder_attention_mask, labels=None):
|
|
|
165 |
self.tokenizer = tokenizer
|
166 |
self.output = output
|
167 |
self.learning_rate = learning_rate
|
168 |
+
self.adam_epsilon = adam_epsilon
|
169 |
self.weight_decay = weight_decay
|
170 |
|
171 |
def forward(self, input_ids, attention_mask, decoder_attention_mask, labels=None):
|