POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

trener c.d.

This commit is contained in:
l.gabrysiak 2025-02-25 13:19:41 +01:00
parent 329d76d072
commit 6d1150308b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hft.py
View File

@ -99,7 +99,7 @@ class CustomModel(AutoModelForCausalLM):
# Dostosowany Trainer
class CustomTrainer(Trainer):
def compute_loss(self, model, inputs, return_outputs=False):
def compute_loss(self, model, inputs, return_outputs=False, num_items_in_batch=None):
labels = inputs.pop("labels")
source = inputs.pop("source")
outputs = model(**inputs, labels=labels)