Skip to content

Instantly share code, notes, and snippets.

@pythonlessons

pythonlessons/transformers_training_13.py

Created September 4, 2023 15:04
Show Gist options
  • Save pythonlessons/67376be380d55b5cfa0a1c55b75028b0 to your computer and use it in GitHub Desktop.
Save pythonlessons/67376be380d55b5cfa0a1c55b75028b0 to your computer and use it in GitHub Desktop.
Download ZIP
transformers_training
Raw
transformers_training_13.py
# Train the model
transformer.fit(
train_dataProvider,
validation_data=val_dataProvider,
epochs=configs.train_epochs,
callbacks=[
warmupCosineDecay,
checkpoint,
tb_callback,
reduceLROnPlat,
model2onnx,
encDecSplitCallback
]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment