Well i think there is a problem with model in particular
but still can you give train a look
lemme add the train loop too in
EDIT -
may there is some problem there
ALSO –
this i cannot interpret this error
TypeError: 'OurAlex' object is not subscriptable