amaniabuzaid commited on
Commit
d7e4f47
·
verified ·
1 Parent(s): f04260c

Update Python

Browse files
Files changed (1) hide show
  1. Python +2 -2
Python CHANGED
@@ -21,7 +21,7 @@ train_data = dataset.map(preprocess_function, batched=True)
21
 
22
  # Define training arguments
23
  training_args = TrainingArguments(
24
- output_dir="./outputs", # Change this to your desired output directory
25
  per_device_train_batch_size=8,
26
  per_device_eval_batch_size=8,
27
  num_train_epochs=3, # Adjust number of epochs for training
@@ -48,4 +48,4 @@ trainer.train()
48
 
49
  # Model is now trained and uploaded to the Hub if push_to_hub was True
50
 
51
- # For manual upload after training, use the Hub API (refer to Hugging Face documentation)
 
21
 
22
  # Define training arguments
23
  training_args = TrainingArguments(
24
+ output_dir="./outputs", # any desired output directory
25
  per_device_train_batch_size=8,
26
  per_device_eval_batch_size=8,
27
  num_train_epochs=3, # Adjust number of epochs for training
 
48
 
49
  # Model is now trained and uploaded to the Hub if push_to_hub was True
50
 
51
+ # For manual upload after training, we use the Hub API (refer to Hugging Face documentation)