search

minyoungg / vqtorch

MIT License
101 stars 9 forks source link

fix bug in residual quantization initialization #7

Closed Jyonn closed 9 months ago

Jyonn commented 9 months ago
  1. self.feature_size is not accessible as the super init is called later
  2. wrong format of d and q. z_q.shape[:-1] -> *z_q.shape[:-2], self.groups
minyoungg commented 9 months ago

Verified that this is indeed a bug. accepting the pull request!