eevaw commited on
Commit
c006e19
·
verified ·
1 Parent(s): dc812ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -79,8 +79,8 @@ torch.cuda.empty_cache()
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,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
- #logging_dir='./logs',
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