stylegan2-flax-tpu icon indicating copy to clipboard operation
stylegan2-flax-tpu copied to clipboard

🖼 Training StyleGAN2 on TPUs in JAX

Results 7 stylegan2-flax-tpu issues
Sort by recently updated
recently updated
newest added

Hi, my end goal is to train the stylegan model using google colab TPU. Right now i'm running the code by following the instructions to first run inference on the...

Hey I read "The image datasets were built using the LAION5B index" That's great, glad the index was useful! I'm curious how you used it exactly, can you provide some...

Without `-p` option command fails with 'no such file or directory' error ![image](https://user-images.githubusercontent.com/14801311/180108127-1a6dc3c9-447f-4a26-9238-4819bb886223.png)

added web demo using gradio on Hugging Face

After running pip to install the requirements, the tensorflow package came with a library that has a incorrect hash. I cant seem to verify the hash either. I haven't changed/updated...

i have a dataset of grayscale images and wanted to train on it. do you have any suggestions on how to get this working with grayscale input images. currently, when...

Hello, I was trying to test and got this error. I am using Colab Pro+ and have a TPU/High Mem. Any idea why this occurs? 2022-12-22 13:40:59.434952: E external/org_tensorflow/tensorflow/compiler/xla/stream_executor/cuda/cuda_driver.cc:267] failed...