Skip to content
GitLab
Commit d0a40969 authored by Мазур Грета Евгеньевна's avatar Мазур Грета Евгеньевна
Browse files

micro zapusk no cross

parent 0fb970ff
master
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 4 additions and 0 deletions
+4 -0
.ipynb_checkpoints/trytoubload-checkpoint.py
+ 2
− 0
View file @ d0a40969
......@@ -14,6 +14,8 @@ from transformers import BertTokenizer, BertPreTrainedModel, BertModel, Trainer,
from torch import nn
from peft import get_peft_model, LoraConfig, TaskType
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
class MultiTaskBert(BertPreTrainedModel):
def __init__(self, config):
......
trytoubload.py
+ 2
− 0
View file @ d0a40969
......@@ -14,6 +14,8 @@ from transformers import BertTokenizer, BertPreTrainedModel, BertModel, Trainer,
from torch import nn
from peft import get_peft_model, LoraConfig, TaskType
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
class MultiTaskBert(BertPreTrainedModel):
def __init__(self, config):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment