(model)
| 12 | from torchmetrics.classification import F1Score |
| 13 | |
| 14 | def freeze_params(model): |
| 15 | for layer in model.parameters(): |
| 16 | layer.requires_grade = False |
| 17 | |
| 18 | class JustificationGenerationModule(pl.LightningModule): |
| 19 |
no outgoing calls
no test coverage detected