rdevon / cortex

A machine learning library for PyTorch
BSD 3-Clause "New" or "Revised" License
92 stars 25 forks source link

inconsistent parameters? #211

Open shizhan1992 opened 4 years ago

shizhan1992 commented 4 years ago

https://github.com/rdevon/cortex/blob/2837b220f9fb73279df3815bb18b274106412c08/cortex/built_ins/models/mine.py#L94 Hi, thank you for this great repository. I am wondering if line 94 and line 111 are correct? shouldn't it be self.mine.routine(generated, generated, Z, Z_P)? because the routine accepts arguments in the order of "X_real, X_fake, Z_real, Z_fake".

Besides, line 123 "self.mine.visualize(generated, generated, Z, Z_N, targets)" seems correct.