batch size 未定义

[yxk9810]

colab 上执行memorylist 报错如下：

frames /usr/local/lib/python3.10/dist-packages/memorag/memorag_lite.py in adapt_batch_size(self) 168 batch_size = bs 169 break --> 170 return batch_size 171 172 def memorize( UnboundLocalError: local variable 'batch_size' referenced before assignment

貌似batch size 未定义

[qhjqhj00]

您是在没有gpu的环境运行吗？
如果是CPU的话，这里有点问题，我尽快修复一下

[yxk9810]

gpu环境