Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
9f5c468
1
Parent(s):
4fd5b35
Update model.py
Browse files- src/models/model.py +0 -1
src/models/model.py
CHANGED
@@ -152,7 +152,6 @@ class PLDataModule(LightningDataModule):
|
|
152 |
class LightningModel(LightningModule):
|
153 |
""" PyTorch Lightning Model class"""
|
154 |
|
155 |
-
def __init__(self, tokenizer, model, learning_rate, adam_epsilon, output: str = "outputs"):
|
156 |
"""
|
157 |
initiates a PyTorch Lightning Model
|
158 |
Args:
|
|
|
152 |
class LightningModel(LightningModule):
|
153 |
""" PyTorch Lightning Model class"""
|
154 |
|
|
|
155 |
"""
|
156 |
initiates a PyTorch Lightning Model
|
157 |
Args:
|