Fix all undefined variables

This commit is contained in:
Max Tretikov 2024-06-13 14:49:00 -06:00
parent 8be5134f4c
commit c69d4cae0a
2 changed files with 3 additions and 1 deletions

View File

@ -1,4 +1,5 @@
import torch
import math
from contextlib import contextmanager
from typing import Any, Dict, List, Optional, Tuple, Union
import logging as logpy
@ -113,7 +114,7 @@ class AutoencodingEngine(AbstractAutoencoder):
self.encoder: torch.nn.Module = instantiate_from_config(encoder_config)
self.decoder: torch.nn.Module = instantiate_from_config(decoder_config)
self.regularization: AbstractRegularizer = instantiate_from_config(
self.regularization: DiagonalGaussianRegularizer = instantiate_from_config(
regularizer_config
)

View File

@ -1,5 +1,6 @@
import logging
import math
from functools import partial
from typing import Dict, Optional
import numpy as np