Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -79,8 +79,8 @@ torch.cuda.empty_cache()
|
|
79 |
#nvidia-smi
|
80 |
|
81 |
#pip install wandb
|
82 |
-
|
83 |
-
|
84 |
from transformers import MT5ForConditionalGeneration, Seq2SeqTrainingArguments, Seq2SeqTrainer
|
85 |
import torch
|
86 |
|
@@ -103,7 +103,7 @@ training_args = Seq2SeqTrainingArguments(
|
|
103 |
per_device_train_batch_size=4, # Pienennä batch-kokoa
|
104 |
per_device_eval_batch_size=4,
|
105 |
evaluation_strategy='epoch',
|
106 |
-
|
107 |
predict_with_generate=True
|
108 |
)
|
109 |
|
|
|
79 |
#nvidia-smi
|
80 |
|
81 |
#pip install wandb
|
82 |
+
import wandb
|
83 |
+
wandb.login(key="1a89092e258a76c696d4cc4338e3ddfb289eff1f")
|
84 |
from transformers import MT5ForConditionalGeneration, Seq2SeqTrainingArguments, Seq2SeqTrainer
|
85 |
import torch
|
86 |
|
|
|
103 |
per_device_train_batch_size=4, # Pienennä batch-kokoa
|
104 |
per_device_eval_batch_size=4,
|
105 |
evaluation_strategy='epoch',
|
106 |
+
logging_dir='./logs',
|
107 |
predict_with_generate=True
|
108 |
)
|
109 |
|