Comments (2)
I used followings to train on FFHQ dataset,
- Refer to FFHQ dataset repo to create
*.tfrecord
for resolution of your interest. - Please use
docker
for training envrionment.- Recommended docker image :
nvcr.io/nvidia/tensorflow:21.03-tf2-py3
- Recommended docker image :
- Train with
train.py
- example 256x256 resolution
# adjust shuffle_buffer_size and batch_size_per_replica to fit your hardware
~$ python train.py \
--allow_memory_growth false \
--use_tf_function true \
--use_custom_cuda true \
--model_base_dir <directory-of-output-trained-model> \
--tfrecord_dir <FFHQ-tfrecord-dir> \
--train_res 256 \
--shuffle_buffer_size 5000 \
--batch_size_per_replica 16
from stylegan2-tf-2.x.
Thank you.
from stylegan2-tf-2.x.
Related Issues (15)
- convert official weights HOT 3
- Cannot convert pkl file to ckpt when the size is 256 or 512. HOT 5
- checkpoint of FFHQ 256*256?
- Error
- when training in 256x256, the output is 256x512 pixels HOT 1
- Memory leak
- 10 errors detected in the compilation of upfirdn_2d.cu
- different resolution pkl file HOT 1
- Please update readme HOT 3
- Training on custom dataset failed HOT 9
- Training crash randomly HOT 2
- What are the "labels" in the generator and discriminator? HOT 2
- Need help understanding the role of the 'upfirdn' after the transpose_conv HOT 1
- NHWC Support HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from stylegan2-tf-2.x.