diff options
| -rw-r--r-- | __pycache__/hyperparameters.cpython-38.pyc | bin | 344 -> 0 bytes | |||
| -rw-r--r-- | __pycache__/losses.cpython-38.pyc | bin | 4620 -> 0 bytes | |||
| -rw-r--r-- | __pycache__/preprocess.cpython-38.pyc | bin | 5074 -> 0 bytes | |||
| -rw-r--r-- | main.py | 10 |
4 files changed, 10 insertions, 0 deletions
diff --git a/__pycache__/hyperparameters.cpython-38.pyc b/__pycache__/hyperparameters.cpython-38.pyc Binary files differdeleted file mode 100644 index 12ab00e4..00000000 --- a/__pycache__/hyperparameters.cpython-38.pyc +++ /dev/null diff --git a/__pycache__/losses.cpython-38.pyc b/__pycache__/losses.cpython-38.pyc Binary files differdeleted file mode 100644 index 1f2c9bf4..00000000 --- a/__pycache__/losses.cpython-38.pyc +++ /dev/null diff --git a/__pycache__/preprocess.cpython-38.pyc b/__pycache__/preprocess.cpython-38.pyc Binary files differdeleted file mode 100644 index c0441c30..00000000 --- a/__pycache__/preprocess.cpython-38.pyc +++ /dev/null @@ -40,9 +40,17 @@ def parse_args(): def train(model): for i in range(hp.num_epochs): + if i % 100 == 0: + fn = f"checkpoint-images/img-epoch{i}.jpg" + save_image(fn, tf.squeeze(model.x)) + print('batch', i) model.train_step() +def save_image(filename, image): + imsave(filename, image) + + def main(): """ Main function. """ if os.path.exists(ARGS.content): @@ -66,6 +74,8 @@ def main(): imsave(ARGS.savefile, final_image) + + ARGS = parse_args() main() |