Skip to content
Snippets Groups Projects

Resolve "Flair Embeddings"

2 files
+ 10
1
Compare changes
  • Side-by-side
  • Inline

Files

+ 9
0
@@ -182,3 +182,12 @@ def predict_model(model, eval_dataset, label_list, batch_size, device, report=Tr
report = classification_report(y_true, y_pred, digits=4)
return y_pred
def countZeroWeights(model):
zeros = 0
for param in model.parameters():
if param is not None:
param = param.cpu()
zeros += param.numel() - param.nonzero().size(0)
return zeros
\ No newline at end of file
Loading