From a8b5d5d753f4b8f88fe1bff6c0929b8663df3421 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Fri, 22 Apr 2022 14:16:43 -0700 Subject: [PATCH] last tweak of readme --- README.md | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/README.md b/README.md index 6084afa..f4cec1a 100644 --- a/README.md +++ b/README.md @@ -485,6 +485,10 @@ with decoder.one_unet_in_gpu(2): loss = decoder(images, unet_number = 2) loss.backward() +with decoder.one_unet_in_gpu(3): + loss = decoder(images, unet_number = 3) + loss.backward() + # do the above for many steps for both unets # then it will learn to generate images based on the CLIP image embeddings