A question about train step

maum-ai / faceshifter

Unofficial PyTorch Implementation for FaceShifter (https://arxiv.org/abs/1912.13457)

BSD 3-Clause "New" or "Revised" License

610 stars 114 forks source link

A question about train step #26

Open minhoooo1 opened 2 years ago

minhoooo1 commented 2 years ago

Excuse me, my training data set has only 1000 pictures, And I found len(AEI_Dataset) = len(self.files) * 5 = 5000 my batch size = 20. So train step should be 250，but why train step is 125000250 。

stau-7001 commented 2 years ago

I also get this problem. Have you solved this problem?