Skip to content

Commit

Permalink
remove some commented out imports that we don't use
Browse files Browse the repository at this point in the history
  • Loading branch information
henryaddison committed Feb 13, 2024
1 parent 450de34 commit 24f53dd
Showing 1 changed file with 2 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -20,25 +20,20 @@
# and MLFlow, and iterating by epoch using PyTorch DataLoaders

# pylint: skip-file
"""Training and evaluation for score-based generative models. """
"""Training for score-based generative models. """

import itertools
import os

from codetiming import Timer
# import tensorflow as tf
# import tensorflow_gan as tfgan
import logging
# Keep the import below for registering all model definitions
# from models import ddpm, ncsnv2, ncsnpp
from .models import cunet
from .models import cncsnpp
from .models import cunet, cncsnpp
from . import losses
from .models.location_params import LocationParams
from . import sampling
from .models import utils as mutils
from .models.ema import ExponentialMovingAverage
# import .evaluation
from . import likelihood
from . import sde_lib
from absl import flags
Expand Down

0 comments on commit 24f53dd

Please sign in to comment.