Skip to content

Instantly share code, notes, and snippets.

@shawwn

shawwn/stylegan2_train_tpu.sh

Last active April 25, 2021 13:44
Show Gist options
  • Star 5 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
  • Save shawwn/890c15427a16c898b299687073938882 to your computer and use it in GitHub Desktop.
Save shawwn/890c15427a16c898b299687073938882 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
stylegan2_train_tpu.sh
#!/bin/bash
# git clone https://github.com/shawwn/stylegan2 -b tpu
source "$HOME/bin/activate-tf1"
set -x
#export TPU_NAME=grpc://0.tcp.ngrok.io:15992
export TPU_NAME=grpc://0.tcp.ngrok.io:17042
export GCE_BUCKET=gs://sgappa-multi/stylegan-encoder
export NOISY=1
export DEBUG=1
export RESUME_PKL=./results/00035-stylegan2-animefaces-1gpu-config-a/network-snapshot-000398.pkl
#exec python3 -m pdb -c continue run_training.py --data-dir ./datasets --config=config-f --dataset=animefaces --mirror-augment=false --metrics=none "$@" # stylegan2
exec python3 -m pdb -c continue run_training.py --data-dir ./datasets --config=config-a --dataset=animefaces --mirror-augment=false --metrics=none "$@" # stylegan1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment