search

GanjinZero / RRHF

[NIPS2023] RRHF & Wombat
780 stars 49 forks source link

loss的代码关于batch size的处理有bug。 #23

Open echoht opened 1 year ago

echoht commented 1 year ago

企业微信截图_16832967678761 这里batch size设置为1时，逻辑没有问题。当batch size！=1时，会出现tensor size不match的情况。

GanjinZero commented 1 year ago

batch size不等于1的时候 一般都会爆显存，所以就没有实现batch size不等于1的情况；感兴趣的话可以自己改一下

echoht commented 1 year ago

batch size不等于1的时候 一般都会爆显存，所以就没有实现batch size不等于1的情况；感兴趣的话可以自己改一下

https://github.com/GanjinZero/RRHF/pull/24/commits/026adca7d79b14a262cca689618d89b57b65e939 帮忙cr下。

GanjinZero commented 1 year ago

下周一我确定代码无误后 再接受pr

echoht commented 1 year ago

下周一我确定代码无误后 再接受pr

https://github.com/GanjinZero/RRHF/pull/24/commits/701aaebc6bc395c67be4c96fe4ecfd03c84eca4a 这个commit哈，是最新的。