Fix #1078

Author: mingxingtan

efficientnetv2/tutorial.ipynb

@@ -84,7 +84,7 @@
         "  ckpt_path = os.path.join(os.getcwd(), m)\n",
         "  return ckpt_path"
       ],
-      "execution_count": 1,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -175,7 +175,7 @@
         "image = tf.io.read_file(image_file)\n",
         "image = preprocessing.preprocess_image(\n",
         "    image, cfg.eval.isize, is_training=False, augname=cfg.data.augname)\n",
-        "logits = model(tf.expand_dims(image, 0), False)[0]\n",
+        "logits = model(tf.expand_dims(image, 0), False)\n",
         "\n",
         "# Output classes and probability\n",
         "pred = tf.keras.layers.Softmax()(logits)\n",
@@ -187,7 +187,7 @@
         "from IPython import display\n",
         "display.display(display.Image(image_file))"
       ],
-      "execution_count": 79,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -242,7 +242,7 @@
       "source": [
         "!python main_tf2.py --mode=traineval  --model_name=efficientnetv2-b0  --dataset_cfg=cifar10Ft --model_dir={MODEL}_finetune --hparam_str=\"train.ft_init_ckpt={MODEL},runtime.strategy=gpus,train.batch_size=64\"\n"
       ],
-      "execution_count": 87,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",