@lampadephoria, I think you should remove override since there is no get function in parent class for you to override.
Take a look at this example:
‘examples/custom-dataset.cpp at master · pytorch/examples · GitHub 3’
Example class is there:
‘pytorch/example.h at master · pytorch/pytorch · GitHub 2’