remove redundant self.eval(), self.train(), and removed def toggle_discretized

This commit is contained in:
jayLEE0301 2024-06-08 15:56:26 -04:00
parent 913489ead0
commit 4d3a45ac26
1 changed files with 1 additions and 1 deletions

View File

@ -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.