[TypeError: batch must contain tensors, numbers, dicts or lists; found object]
I met exactly same error and I found that the type of data I create in  init  is object. I specified it to float and solved my problem.