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 |
|