oasis:

for the next part, I mean if only passing in one image, the Net won’t work and will return ’ [Error: Expected more than 1 value per channel when training]’


I guess this is because you have a batchnorm or similar normalization layer in there that can only work in training mode when you pass in more than 1 sample?