train_dataloader.batch_size = Unfreeze_batch_size val_dataloader.batch_size = Unfreeze_batch_size
时间: 2024-01-24 14:19:08 浏览: 173
train_data.txt
这句话是在设置训练集和验证集的 batch size,其中 train_dataloader.batch_size 和 val_dataloader.batch_size 都被设置为 Unfreeze_batch_size。这个 batch size 是在模型 fine-tuning 时解冻所有层进行训练时使用的 batch size。
阅读全文