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

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