manual fixes for some C408

This commit is contained in:
AUTOMATIC
2023-05-10 11:55:09 +03:00
parent a5121e7a06
commit d25219b7e8
7 changed files with 15 additions and 15 deletions

View File

@@ -237,7 +237,7 @@ class VQModel(pl.LightningModule):
return self.decoder.conv_out.weight
def log_images(self, batch, only_inputs=False, plot_ema=False, **kwargs):
log = dict()
log = {}
x = self.get_input(batch, self.image_key)
x = x.to(self.device)
if only_inputs: