Skip to content

Commit 9429c3e

Browse files
committed
Remove capture from notebooks/influence_imagenet and use tag to hide output
1 parent 9840d70 commit 9429c3e

File tree

2 files changed

+46
-10
lines changed

2 files changed

+46
-10
lines changed

mkdocs.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,8 @@ plugins:
5353
- hide
5454
remove_input_tags:
5555
- hide-input
56+
remove_all_outputs_tags:
57+
- hide-output
5658
binder: true
5759
binder_service_name: "gh"
5860
binder_branch: "develop"

notebooks/influence_imagenet.ipynb

Lines changed: 44 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,13 @@
22
"cells": [
33
{
44
"cell_type": "markdown",
5-
"metadata": {},
5+
"metadata": {
6+
"editable": true,
7+
"slideshow": {
8+
"slide_type": ""
9+
},
10+
"tags": []
11+
},
612
"source": [
713
"# Influence functions for neural networks\n",
814
"\n",
@@ -261,10 +267,17 @@
261267
{
262268
"cell_type": "code",
263269
"execution_count": 7,
264-
"metadata": {},
270+
"metadata": {
271+
"editable": true,
272+
"slideshow": {
273+
"slide_type": ""
274+
},
275+
"tags": [
276+
"hide-output"
277+
]
278+
},
265279
"outputs": [],
266280
"source": [
267-
"%%capture\n",
268281
"model_ft = new_resnet_model(output_size=len(label_names))\n",
269282
"mgr = TrainingManager(\n",
270283
" \"model_ft\",\n",
@@ -376,10 +389,17 @@
376389
{
377390
"cell_type": "code",
378391
"execution_count": 10,
379-
"metadata": {},
392+
"metadata": {
393+
"editable": true,
394+
"slideshow": {
395+
"slide_type": ""
396+
},
397+
"tags": [
398+
"hide-output"
399+
]
400+
},
380401
"outputs": [],
381402
"source": [
382-
"%%capture\n",
383403
"influences = compute_influences(\n",
384404
" TorchTwiceDifferentiable(mgr.model, mgr.loss),\n",
385405
" train_data,\n",
@@ -662,10 +682,17 @@
662682
{
663683
"cell_type": "code",
664684
"execution_count": 17,
665-
"metadata": {},
685+
"metadata": {
686+
"editable": true,
687+
"slideshow": {
688+
"slide_type": ""
689+
},
690+
"tags": [
691+
"hide-output"
692+
]
693+
},
666694
"outputs": [],
667695
"source": [
668-
"%%capture\n",
669696
"corrupted_model = new_resnet_model(output_size=len(label_names))\n",
670697
"corrupted_dataset, corrupted_indices = corrupt_imagenet(\n",
671698
" dataset=train_ds,\n",
@@ -756,10 +783,17 @@
756783
{
757784
"cell_type": "code",
758785
"execution_count": 20,
759-
"metadata": {},
786+
"metadata": {
787+
"editable": true,
788+
"slideshow": {
789+
"slide_type": ""
790+
},
791+
"tags": [
792+
"hide-output"
793+
]
794+
},
760795
"outputs": [],
761796
"source": [
762-
"%%capture\n",
763797
"influences = compute_influences(\n",
764798
" TorchTwiceDifferentiable(mgr.model, mgr.loss),\n",
765799
" corrupted_data,\n",
@@ -1060,5 +1094,5 @@
10601094
}
10611095
},
10621096
"nbformat": 4,
1063-
"nbformat_minor": 2
1097+
"nbformat_minor": 4
10641098
}

0 commit comments

Comments
 (0)