TransGAN

KO | EN

gitlite — search

#javascript #ai #typescript #golang #go #android #machine-learning #rust #linux #nodejs #kubernetes #security #mcp #open-source

(★ 25)

Implementation of Transformer-based GAN Model in Tensorflow / Keras

TransGAN 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

## 🔥 News
- [2024-09-13]  The new Transformer GAN model, [LadaGAN](https://github.com/milmor/LadaGAN), has been released. It offers improved FID evaluation results, includes model checkpoints, and requires only a single GPU for training. The code has been optimized for better performance and now offers additional functionalities.

# TransGAN
Implementation of the _Transformer-based GAN_ model in the paper:

> [TransGAN: Two Pure Transformers Can Make One
Strong GAN, and That Can Scale Up](https://arxiv.org/abs/2102.07074).

![Architecture](https://raw.githubusercontent.com/milmor/TransGAN/main/images/architecture.png)

See [here](https://github.com/VITA-Group/TransGAN) for the official Pytorch implementation.

## Dependencies
- Python 3.8
- Tensorfow 2.5

## Usage
### Train
1. Use `--dataset_path=<path>` to specify the dataset path (default builds CIFAR-10 dataset), and `--model_name=<name>` to specify the checkpoint directory name.
```
python train.py --dataset_path=<path> --model_name=<name> 
```

### Hparams setting
Adjust hyperparameters in the `hparams.py` file.

### Tensorboard
Run `tensorboard --logdir ./`.

## Examples
- CIFAR-10 training progress

![](https://raw.githubusercontent.com/milmor/TransGAN/main/images/transgan_samples.gif "TransGAN on CIFAR-10")

## References
Code:
- This model depends on other files that may be licensed under different open source licenses.
- TransGAN uses [Differentiable Augmentation](https://arxiv.org/abs/2006.10738). Under BSD 2-Clause "Simplified" License.
- Small-TransGAN models are instances of the original TransGAN architecture with a smaller number of layers and lower-dimensional embeddings.

Implementation notes:
- Single layer per resolution Generator.
- Orthogonal initializer and 4 heads in both Generator and Discriminator.
- WGAN-GP loss.
- Adam with β1 = 0.0 and β2 = 0.99.
- Noise dimension = 64.
- Batch size = 64

## Licence
MIT

TransGAN

TransGAN 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

TransGAN

TransGAN 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README