You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thank you for reverse engineering this paper. When I tried to run python train_mac.py I got the following error:
Traceback (most recent call last):
File "/content/titans-pytorch/train_mac.py", line 75, in <module>
model = MemoryAsContextTransformer(
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/content/titans-pytorch/titans_pytorch/mac_transformer.py", line 543, in __init__
mem = NeuralMemory(
^^^^^^^^^^^^^
TypeError: NeuralMemory.__init__() got an unexpected keyword argument 'learned_mem_model_weights'
I fixed it by commenting 'learned_mem_model_weights = LEARNED_MEM_MODEL_WEIGHTS,' out of:
Thank you for reverse engineering this paper. When I tried to run python train_mac.py I got the following error:
I fixed it by commenting 'learned_mem_model_weights = LEARNED_MEM_MODEL_WEIGHTS,' out of:
Not sure if it is important.
Thanks.
The text was updated successfully, but these errors were encountered: