Michał Marcińczuk
--- a/poldeepner2/model/herbet_for_token_calssification.py 0 → 100644
+++ b/poldeepner2/model/herbet_for_token_calssification.py 0 → 100644
+from transformers import AutoModel, AutoTokenizer
+import torch.nn as nn
+import torch.nn.functional as F
+import torch
+
+class HerbertForTokenClassification(nn.Module):
+
+    def __init__(self, pretrained_path, n_labels, hidden_size=768, dropout_p=0.2, label_ignore_idx=0,