diff --git a/notebooks/huggingface_notebook.ipynb b/notebooks/huggingface_notebook.ipynb index 799fb84b33..9b813f8c7d 100644 --- a/notebooks/huggingface_notebook.ipynb +++ b/notebooks/huggingface_notebook.ipynb @@ -134,7 +134,7 @@ " nb_classes=10,\n", " clip_values=(0, 1),\n", " processor=upsampler)\n", - " hf_model.fit(x_train, y_train, nb_epochs=2)\n", + " hf_model.fit(x_train, y_train, nb_epochs=2, display_progress_bar=True)\n", " return hf_model" ] }, @@ -161,6 +161,48 @@ "- classifier.weight: found shape torch.Size([1000, 768]) in the checkpoint and torch.Size([10, 768]) in the model instantiated\n", "You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.\n" ] + }, + { + "data": { + "application/vnd.jupyter.widget-view+json": { + "model_id": "7794a0d05b504c48a8d70d1c0596e3a1", + "version_major": 2, + "version_minor": 0 + }, + "text/plain": [ + "Epochs: 0%| | 0/2 [00:00" ] @@ -414,7 +456,7 @@ { "data": { "application/vnd.jupyter.widget-view+json": { - "model_id": "fbb7dfd7f7434b809d0dbc1252f49bd1", + "model_id": "", "version_major": 2, "version_minor": 0 }, @@ -480,6 +522,48 @@ "- classifier.weight: found shape torch.Size([1000, 768]) in the checkpoint and torch.Size([10, 768]) in the model instantiated\n", "You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.\n" ] + }, + { + "data": { + "application/vnd.jupyter.widget-view+json": { + "model_id": "d1e1c1da709f43279594cebafaf48077", + "version_major": 2, + "version_minor": 0 + }, + "text/plain": [ + "Epochs: 0%| | 0/2 [00:00" ] @@ -617,7 +701,7 @@ " clip_values=(0, 1),\n", " processor=upsampler)\n", " (x_train, y_train), (x_test, y_test) = get_cifar_data(train_on_subset)\n", - " hf_model.fit(x_train, y_train, nb_epochs=2)\n", + " hf_model.fit(x_train, y_train, nb_epochs=2, display_progress_bar=True)\n", "\n", " return hf_model" ] @@ -635,6 +719,48 @@ "Files already downloaded and verified\n", "Files already downloaded and verified\n" ] + }, + { + "data": { + "application/vnd.jupyter.widget-view+json": { + "model_id": "eceb751607a642da8c06b35bb2ea1aba", + "version_major": 2, + "version_minor": 0 + }, + "text/plain": [ + "Epochs: 0%| | 0/2 [00:00