Restored old components

config.py

@@ -61,7 +61,7 @@ def from_args(self, args=sys.argv[1:]):
         parser.add_argument('--video', dest='make_video', action='store_true', help='Generate output videos (MP4s)')
         parser.add_argument('--batch', dest='batch_mode', action='store_true', help="Don't open windows, instead save results to file")
         parser.add_argument('-b', dest='batch_size', type=int, default=None, help='Minibatch size, leave empty for automatic detection')
-        parser.add_argument('-c', dest='components', type=int, default=128, help='Number of components to keep')
+        parser.add_argument('-c', dest='components', type=int, default=80, help='Number of components to keep')
         parser.add_argument('-n', type=int, default=300_000, help='Number of examples to use in decomposition')
         parser.add_argument('--use_w', action='store_true', help='Use W latent space (StyleGAN(2))')
         parser.add_argument('--sigma', type=float, default=2.0, help='Number of stdevs to walk in visualize.py')

notebooks/figure_teaser.ipynb

@@ -219,7 +219,8 @@
    "outputs": [],
    "source": [
     "# BigGAN-512 irish setter\n",
-    "setup_model('BigGAN-512', 'irish_setter', 'generator.gen_z')\n",
+    "setup_model('BigGAN-512', 'husky', 'generator.gen_z')\n",
+    "model.set_output_class('irish_setter')\n",
     "model.truncation = 0.6\n",
     "\n",
     "seeds = [489408325]# + [rand() for _ in range(10)]\n",
@@ -251,9 +252,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.5-final"
+   "version": "3.7.7"
   }
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}