search

facebookresearch / deepcluster

Deep Clustering for Unsupervised Learning of Visual Features
Other
1.69k stars 325 forks source link

Return statement inside for loop cuts short each set of epochs. #1

Closed kylehkhsu closed 6 years ago

kylehkhsu commented 6 years ago

I believe that the return statement found here should be de-indented. As-is, the model trains for only a single batch between clusterings.

mathildecaron31 commented 6 years ago

Hi, Yes indeed, thank you for pointing this out !