XuezheMax / wolf

Invertible Generative Flows
Apache License 2.0
81 stars 13 forks source link

'tiny change' #1

Closed plutoyuxie closed 3 years ago

plutoyuxie commented 3 years ago
  1. Modifying tiny mistake in README.md.

  2. Apex might be optional for who has difficulty installing this package?

  3. "RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation" error occurred. My device is a 3090 GPU with pytorch 1.7. I change scale = log_scale.mul_(0.5).tanh_().mul(self.alpha).add(1.0) to scale = log_scale.clone().mul_(0.5).tanh_().mul(self.alpha).add(1.0) and this error disappeared. Still not sure if it was correct.