benchmarking.benchmark_definitions.finetune_transformer_swag module

benchmarking.benchmark_definitions.finetune_transformer_swag.finetune_transformer_swag_benchmark(sagemaker_backend=False, num_train_epochs=3, per_device_train_batch_size=8, **kwargs)

Parameters:

• sagemaker_backend (bool) – Use SageMaker backend? This affects the choice of instance type. Defaults to False

• num_train_epochs (int) – Maximum number of epochs for fine-tuning. Defaults to 3

• per_device_train_batch_size (int) – Batch size per device. Defaults to 8

• kwargs – Overwrites default params in RealBenchmarkDefinition object returned

Return type:

RealBenchmarkDefinition