Add TensorFlow checkpoint to PyTorch converter#8

Open

jesse1029 wants to merge 1 commit into

codex/convert-to-pytorch-version-91c39t

jesse1029 commented Sep 29, 2025

Owner

Summary

align the PyTorch DCGAN generator and discriminator with the TensorFlow layout to allow weight imports
add a conversion utility that loads TensorFlow checkpoints and saves matching PyTorch state_dict files
document how to run the checkpoint converter alongside the existing PyTorch training workflow

Testing

python -m py_compile models_64x64_torch.py train_celeba_dcgan_torch.py convert_tf_checkpoint_to_torch.py

https://chatgpt.com/codex/tasks/task_e_68da199bb25083299f8ebcee3a091df4


          Add TensorFlow to PyTorch DCGAN checkpoint converter

e4dda7b

jesse1029 added the codex label

— with

ChatGPT Codex Connector

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels