Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
1cc2363
1
Parent(s):
ac05d72
Update model.py
Browse files- src/models/model.py +0 -1
src/models/model.py
CHANGED
|
@@ -161,7 +161,6 @@ class LightningModel(LightningModule):
|
|
| 161 |
self.model = model
|
| 162 |
self.tokenizer = tokenizer
|
| 163 |
self.output = output
|
| 164 |
-
# self.val_acc = Accuracy()
|
| 165 |
# self.train_acc = Accuracy()
|
| 166 |
|
| 167 |
def forward(self, input_ids, attention_mask, decoder_attention_mask, labels=None):
|
|
|
|
| 161 |
self.model = model
|
| 162 |
self.tokenizer = tokenizer
|
| 163 |
self.output = output
|
|
|
|
| 164 |
# self.train_acc = Accuracy()
|
| 165 |
|
| 166 |
def forward(self, input_ids, attention_mask, decoder_attention_mask, labels=None):
|