Downloads · 30 days
14
4% of all-time downloads
bogdancazan/bart-base-newsela-biendata-with-domain-adaptation
bart-base-newsela-biendata-with-domain-adaptation is a machine learning model from bogdancazan. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
trainingargs = TrainingArguments( outputdir='bart-base-newsela-biendata-with-domain-adaptation', numtrainepochs=20, warmupsteps=250, perdevicetrainbatchsize=BATCHSIZE, weightdecay=0.01, learningrate=2e-4, fp16=True, o…
Downloads · 30 days
14
4% of all-time downloads
All-time downloads
347
Public
Repo size
1.7 GB
Likes
0
Public
Click a slice to open those files.
.bin558 MB · 99%
From the Hugging Face model README
training_args = TrainingArguments( output_dir='bart-base-newsela-biendata-with-domain-adaptation', num_train_epochs=20, warmup_steps=250, per_device_train_batch_size=BATCH_SIZE, weight_decay=0.01, learning_rate=2e-4, fp16=True, optim="adafactor", )
Step Training Loss 500 5.677000 1000 2.361900 1500 1.826000 2000 1.672900 2500 1.597900 3000 1.555700 3500 1.520600 4000 1.496300 4500 1.476800 TrainOutput(global_step=4640, training_loss=2.1116079396214977, metrics={'train_runtime': 1059.6025, 'train_samples_per_second': 279.992, 'train_steps_per_second': 4.379, 'total_flos': 0.0, 'train_loss': 2.1116079396214977, 'epoch': 20.0})