Hi,
I cannot trace the code to find out what kind of criterion you have employed for training phase.
To be more specific, I am trying to finetune your model on my own data, However I'm encountering the following error, which is related to criterion part:
/home/amir/torch/install/share/lua/5.1/nn/THNN.lua:110: bad argument #2 to 'v' (mismatch between the batch size of input (1) and that of target (2) at /tmp/luarocks_cunn-scm-1-7242/cunn/lib/THCUNN/generic/ClassNLLCriterion.cu:41)
I checked the size of input and target they are the same.
Thank you in advance.
Hi,
I cannot trace the code to find out what kind of criterion you have employed for training phase.
To be more specific, I am trying to finetune your model on my own data, However I'm encountering the following error, which is related to criterion part:
/home/amir/torch/install/share/lua/5.1/nn/THNN.lua:110: bad argument #2 to 'v' (mismatch between the batch size of input (1) and that of target (2) at /tmp/luarocks_cunn-scm-1-7242/cunn/lib/THCUNN/generic/ClassNLLCriterion.cu:41)
I checked the size of input and target they are the same.
Thank you in advance.