Update Python
Browse files
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", #
|
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)
|