flytoe commited on
Commit
65533e2
·
verified ·
1 Parent(s): 2a2af5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -45,8 +45,7 @@ trainer = Trainer(
45
  model=model,
46
  args=training_args,
47
  train_dataset=dataset["train"],
48
- eval_dataset=dataset["validation"],
49
- compute_loss=lambda model, inputs: model(**inputs).loss # Fix für fehlende Loss-Berechnung
50
  )
51
  trainer.train()
52
 
 
45
  model=model,
46
  args=training_args,
47
  train_dataset=dataset["train"],
48
+ eval_dataset=dataset["validation"]
 
49
  )
50
  trainer.train()
51