有高手知道pytorch如何指定和应用样本权重么

水木社区手机版

主题:有高手知道pytorch如何指定和应用样本权重么
楼主|jesseli|2022-10-12 09:04:39|展开
keras 直接用fit函数的sample_weight参数可以指定训练集样本权重，validation_data中也可以指定验证集样本权重。
pytorch 没找到
--
FROM 120.244.86.*
2楼|jesseli|2022-10-14 23:04:48|展开
keras是tensorflow的高级封装， pytorch的高级封装我用的是pytorch-lightning, 没找到指定样本权重的地方。
自定义loss应该能实现，但是不如框架本身就封装了省功夫。
【在 RuralHunter 的大作中提到: 】
: pytorch不也有keras吗？
:
: #发自zSMTH@Fish Boat
--
FROM 120.244.86.*