Hello @dymat ! Can you please tell me how much time a single iteration is taking and a whole epoch? Did you tried on some more sophisticated dataset or images?