Skip to content

Commit 2bab2a5

Browse files
committed
remove task type from model notebooks
1 parent e6eb05a commit 2bab2a5

File tree

1 file changed

+4
-22
lines changed

1 file changed

+4
-22
lines changed

notebooks/200_models/201_fastflow.ipynb

+4-22
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,6 @@
7575
"from PIL import Image\n",
7676
"from torch.utils.data import DataLoader\n",
7777
"\n",
78-
"from anomalib import TaskType\n",
7978
"from anomalib.data import MVTec, PredictDataset\n",
8079
"from anomalib.engine import Engine\n",
8180
"from anomalib.models import Fastflow\n",
@@ -93,18 +92,7 @@
9392
"source": [
9493
"## Data Module\n",
9594
"\n",
96-
"To train the model end-to-end, we do need to have a dataset. In our [previous notebooks](https://github.com/openvinotoolkit/anomalib/tree/main/notebooks/100_datamodules), we demonstrate how to initialize benchmark- and custom datasets. In this tutorial, we will use MVTec AD DataModule. We assume that `datasets` directory is created in the `anomalib` root directory and `MVTec` dataset is located in `datasets` directory.\n",
97-
"\n",
98-
"Before creating the dataset, let's define the task type that we will be working on. In this notebook, we will be working on a segmentation task. Therefore the `task` variable would be:\n"
99-
]
100-
},
101-
{
102-
"cell_type": "code",
103-
"execution_count": null,
104-
"metadata": {},
105-
"outputs": [],
106-
"source": [
107-
"task = TaskType.SEGMENTATION"
95+
"To train the model end-to-end, we do need to have a dataset. In our [previous notebooks](https://github.com/openvinotoolkit/anomalib/tree/main/notebooks/100_datamodules), we demonstrate how to initialize benchmark- and custom datasets. In this tutorial, we will use MVTec AD DataModule. We assume that `datasets` directory is created in the `anomalib` root directory and `MVTec` dataset is located in `datasets` directory.\n"
10896
]
10997
},
11098
{
@@ -123,7 +111,6 @@
123111
" train_batch_size=32,\n",
124112
" eval_batch_size=32,\n",
125113
" num_workers=0,\n",
126-
" task=task,\n",
127114
")"
128115
]
129116
},
@@ -541,7 +528,7 @@
541528
],
542529
"metadata": {
543530
"kernelspec": {
544-
"display_name": "anomalib",
531+
"display_name": ".venv",
545532
"language": "python",
546533
"name": "python3"
547534
},
@@ -555,14 +542,9 @@
555542
"name": "python",
556543
"nbconvert_exporter": "python",
557544
"pygments_lexer": "ipython3",
558-
"version": "3.11.8"
545+
"version": "3.10.14"
559546
},
560-
"orig_nbformat": 4,
561-
"vscode": {
562-
"interpreter": {
563-
"hash": "f26beec5b578f06009232863ae217b956681fd13da2e828fa5a0ecf8cf2ccd29"
564-
}
565-
}
547+
"orig_nbformat": 4
566548
},
567549
"nbformat": 4,
568550
"nbformat_minor": 2

0 commit comments

Comments
 (0)