Update train.py

train.py

@@ -21,13 +21,12 @@ tokenized_dataset = dataset["train"].map(tokenize_function)
 # Training arguments
 training_args = TrainingArguments(
     output_dir="trained_model",
-    evaluation_strategy="no",
     learning_rate=2e-5,
     per_device_train_batch_size=4,
     num_train_epochs=5,
     weight_decay=0.01,
     save_total_limit=1,
-    logging_dir="./logs",
+    logging_steps=1
 )
 
 # Trainer