{ | |
'model_name':'distilroberta-base', | |
'n_labels':len(attri), | |
'bs':128, | |
'lr': 1.5e-6, | |
'warmup':0.2, | |
'train_size':len(ucc_data_module.train_dataloader()), | |
'w_decay':0.001, | |
'n_epochs':100 | |
} | |
{ | |
'model_name':'distilroberta-base', | |
'n_labels':len(attri), | |
'bs':128, | |
'lr': 1.5e-6, | |
'warmup':0.2, | |
'train_size':len(ucc_data_module.train_dataloader()), | |
'w_decay':0.001, | |
'n_epochs':100 | |
} | |