Closed ssbilakeri closed 2 years ago
self.nl_can = NONLocalBlock2D(in_channels=hidden_dim) NameError: name 'NONLocalBlock2D' is not defined
Hi,
sorry about that. This line is never actually used. I'll remove it from the repo.
self.nl_can = NONLocalBlock2D(in_channels=hidden_dim) NameError: name 'NONLocalBlock2D' is not defined