Sorry to interrupt you again.

[NabsTao]

Thank you for your kind cooperation.

My data set includes two classes of pictures, both of them contains 660 pictures.
If I use your code to train the model, which places need to be modified?

for step in range(150): Need I change this to for step in range(1320):

And someshere that I need to change.