search

bojone / SimCSE

SimCSE在中文任务上的简单实验
591 stars 83 forks source link

怎么修改batch_size #4

Open yuxuan2015 opened 3 years ago

yuxuan2015 commented 3 years ago

企业微信截图_16238418703954

bojone commented 3 years ago 
train_generator = data_generator(train_token_ids, 64)

里边的64就是batch_size