search

THUDM / SwissArmyTransformer

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.
https://THUDM.github.io/SwissArmyTransformer
Apache License 2.0
951 stars 90 forks source link

请问针对样本数量不均衡的数据集怎么做样本均衡呢 #163

Open lln556 opened 7 months ago

1049451037 commented 7 months ago

sat的args里支持传入一个列表的数据集，通过weight可以调整每个数据集的权重：

--train-data data1 data2 data3
--train-data-weights 5 2 3