Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

why pytorch version uses tensorflow? #60

Open
scut-ken5 opened this issue Jun 6, 2024 · 1 comment
Open

why pytorch version uses tensorflow? #60

scut-ken5 opened this issue Jun 6, 2024 · 1 comment

Comments

@scut-ken5
Copy link

The code below is from run_lib.py. I just wanna ask why there exists the tensorflow in the pytorch version of score_sde. Thanks for your answer
import numpy as np import tensorflow as tf import tensorflow_gan as tfgan import logging from models import ddpm, ncsnv2, ncsnpp import losses import sampling from models import utils as mutils from models.ema import ExponentialMovingAverage import datasets import evaluation import likelihood import sde_lib from absl import flags import torch from torch.utils import tensorboard from torchvision.utils import make_grid, save_image from utils import save_checkpoint, restore_checkpoint

@johndpope
Copy link

at quick glance -
this paper originated from google / tensorflow - so it seems the training / eval test sets that are baked into tensorflow got ported across.

"""Return training and evaluation/test datasets from config files."""
import jax
import tensorflow as tf
import tensorflow_datasets as tfds

plus tensorboard

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants