（WaterGAN）TypeError: Expect argument to be a string or int, found ＜class ‘float‘＞

运行WaterGAN出现错误：

TypeError: Expect argument to be a string or int, found <class 'float'>

将mainmhl.py中：

flags.DEFINE_integer("train_size", np.inf, "The size of train images [np.inf]")

改成：

flags.DEFINE_float("train_size", np.inf, "The size of train images [np.inf]")