前提
要确保自己的conda虚拟环境中已经安装了TensorBoard包,如果没有请在终端自行install。
1.定义TrainingArguments
通过logging_dir
参数制定logging日志输出的位置,logging_steps
表示logging记录的步长。
training_args = TrainingArguments(
output_dir='./results', # output directory
num_train_epochs=50, # total number of training epochs
per_device_train_batch_size=256,