KBOBERT Pre-Training 코드 구현
accuracy : 72.9000%
training_args = TrainingArguments(
output_dir='./results',
save_total_limit=5,
save_steps=1000,
num_train_epochs=10,
learning_rate=1e-5,
per_device_train_batch_size=4,
per_device_eval_batch_size=128,
warmup_steps=500,
weight_decay=0.1,
logging_dir='./logs',
logging_steps=100,
evaluation_strategy='steps',
eval_steps = 1000,
)
training_args = TrainingArguments(
output_dir='./results',
save_total_limit=5,
save_steps=1000,
num_train_epochs=5,
learning_rate=1e-5,
per_device_train_batch_size=4,
per_device_eval_batch_size=128,
warmup_steps=1000,
weight_decay=0.1,
logging_dir='./logs',
logging_steps=100,
evaluation_strategy='steps',
eval_steps = 1000,
)
오늘보다 더 성장한 내일의 저를 기대하며, 다음 주에 뵙도록 하겠습니다.
읽어주셔서 감사합니다!