I was able to fix this for the moment by hardcoding hyperparameters as @bwengals suggested. However, I still get errors if I try to train with a different dataset. For the moment this works for me. Thanks.
I was able to fix this for the moment by hardcoding hyperparameters as @bwengals suggested. However, I still get errors if I try to train with a different dataset. For the moment this works for me. Thanks.