remove redundant self.eval(), self.train(), and removed def toggle_discretized
This commit is contained in:
parent
913489ead0
commit
4d3a45ac26
|
@ -631,7 +631,7 @@ class VQBeTScheduler(nn.Module):
|
|||
self.lr_scheduler.step()
|
||||
|
||||
class VQBeTRgbEncoder(nn.Module):
|
||||
"""Encoder an RGB image into a 1D feature vector.
|
||||
"""Encode an RGB image into a 1D feature vector.
|
||||
|
||||
Includes the ability to normalize and crop the image first.
|
||||
|
||||
|
|
Loading…
Reference in New Issue