#!/bin/bash

# a simple script run stylegan2_pytorch on a linode GPU instance

# PLACE=/tmp/neural/
# mkdir -p $PLACE
# cd $PLACE

echo "using stlyeGAN2..."

phrases=("An energetic nomad on a long hike with wooden staff a green leaf and a small dog." "At the beginning of a magical quest for wisdom an animus of a conman hide beneath the table." "A charismatic female figure meditates on the book of flesh in isolation, oval and white." "Fertility of nature in small green leaves, adam's apple, a pyramid and a baby eagle's wing." "The alchemical gold adorns the throne in a powerful house rooted in matter.")

echo "cosmic entanglement..."

cd /root

# transfer learning using ffhq model for sytleGAN2

stylegan2_pytorch --name flickr3 \
		  --data data/1024x1024 \
		  --models_dir models/ffhq-foam \
		  --results_dir results/flickr3 \
		  --multi-gpus \
		  --batch-size 16 --network-capacity 64 \
		  --aug-prob 0.25 --attn-layers 1

# note in big-sleep.py -> torch,cuda.set_device(1)
#  and/or CUDA_VISIBLE_DEVICES=1,2 python myscript.py


# ffmpeg -r 24 -f image2 -s 1920x1080 -i The_Grand_Evocateur_of_intensity.%d.png -vcodec libx264 -crf 25  -pix_fmt yuv420p The_Grand_Evocateur_of_intensity_m3.mp4