init

app.py

@@ -32,12 +32,12 @@ def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr
         training_args = TrainingArguments(
             output_dir='./results',
             evaluation_strategy="epoch",
-            learning_rate=lr,
+            learning_rate=lr**-5,
             per_device_train_batch_size=batch_size,
             per_device_eval_batch_size=batch_size, 
             num_train_epochs=num_epochs,
             weight_decay=0.01,
-            gradient_accumulation_steps=grad,
+            gradient_accumulation_steps=grad**-1,
             load_best_model_at_end=True,
             metric_for_best_model="accuracy",
             greater_is_better=True,