Please check the colab notebook
Closed this issue · 4 comments
Hello,
The notebook was running fine 2 months ago, but no longer after the huggingface update.
I think there is a problem with the load_data_from_folder function, because I get the following error message when training:
AxisError: axis 1 is out of bounds for array of dimension 1 (see pic.).
Thanks for helping!
I am adding longformer support to the Multimodal Toolkit (see pull request section for details). The colab notebook is now completing 1 epoch and I am receiving the same error that Aymen-lab describes above. I've had issues in the past with the calc_classification_metrics(p: EvalPrediction) section and it seems like this is the step where the error is occurring.
Hi @jtfields ,
Thank you for your reply. I hope this problem get solved soon.
I will keep tracking your pull request :)
@Aymen-lab I think this might be related to Issue #41. Look at the code I posted for this error.
Hi @Aymen-lab,
John is on the number with this one. I've made a PR #44 to fix this, but feel free to just copy the relevant block of code from the PR or John's solution above to handle this for now.