Skip to content

Commit 13a7609

Browse files
ashahbaashahba
committed
Prepare for v0.4 release
Signed-off-by: Abolfazl Shahbazi <[email protected]>
2 parents 6a05136 + a7ab339 commit 13a7609

File tree

126 files changed

+6451
-1565
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

126 files changed

+6451
-1565
lines changed

‎.dockerignore‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
.git/**
2+
Dockerfile*

‎.gitignore‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,5 @@ _autosummary
1414
.coverage
1515
**/*.cache
1616
**/**venv
17+
**/**/models/
18+
.vscode

‎Dockerfile‎

Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,74 @@
1+
#
2+
# Copyright (c) 2023 Intel Corporation
3+
#
4+
# Licensed under the Apache License, Version 2.0 (the "License");
5+
# you may not use this file except in compliance with the License.
6+
# You may obtain a copy of the License at
7+
#
8+
# http://www.apache.org/licenses/LICENSE-2.0
9+
#
10+
# Unless required by applicable law or agreed to in writing, software
11+
# distributed under the License is distributed on an "AS IS" BASIS,
12+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
# See the License for the specific language governing permissions and
14+
# limitations under the License.
15+
#
16+
# SPDX-License-Identifier: Apache-2.0
17+
18+
ARG IMAGE_NAME=ubuntu
19+
ARG IMAGE_TAG=22.04
20+
FROM ${IMAGE_NAME}:${IMAGE_TAG} as base
21+
22+
# TLT base target
23+
FROM base as tlt-base
24+
25+
ARG PYTHON=python3
26+
27+
ENV DEBIAN_FRONTEND=noninteractive
28+
29+
RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
30+
libgl1 \
31+
libglib2.0-0 \
32+
${PYTHON} \
33+
${PYTHON}-pip && \
34+
apt-get clean autoclean && \
35+
apt-get autoremove -y && \
36+
rm -rf /var/lib/apt/lists/*
37+
38+
RUN ln -sf $(which ${PYTHON}) /usr/bin/python
39+
40+
# TLT target for development
41+
FROM tlt-base as tlt-devel
42+
43+
ENV DEBIAN_FRONTEND=noninteractive
44+
45+
ENV LANG C.UTF-8
46+
ARG PYTHON=python3
47+
48+
RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
49+
${PYTHON}-dev \
50+
${PYTHON}-distutils \
51+
build-essential \
52+
ca-certificates \
53+
make \
54+
pandoc && \
55+
apt-get clean autoclean && \
56+
apt-get autoremove -y && \
57+
rm -rf /var/lib/apt/lists/*
58+
59+
COPY . /tmp/intel-transfer-learning
60+
61+
WORKDIR /tmp/intel-transfer-learning
62+
63+
RUN ${PYTHON} setup.py bdist_wheel && \
64+
pip install -f https://download.pytorch.org/whl/cpu/torch_stable.html --no-cache-dir dist/*.whl && \
65+
pip install --no-cache-dir 'tensorflow-text==2.11.0'
66+
67+
# TLT target for deployment
68+
FROM tlt-base as tlt-prod
69+
70+
COPY --from=tlt-devel /usr/local/lib/python3.10/dist-packages /usr/local/lib/python3.10/dist-packages
71+
COPY --from=tlt-devel /usr/local/bin /usr/local/bin
72+
73+
ENV DATASET_DIR=/tmp/data
74+
ENV OUTPUT_DIR=/tmp/output

‎Makefile‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ tlt_test_venv: $(CURDIR)/tests/requirements-test.txt
3535
@echo "Building the TLT API in tlt_test_venv env..."
3636
@. $(ACTIVATE_TEST_VENV) && pip install --editable .
3737
@echo "Required for TensorFlow text classification..."
38-
@. $(ACTIVATE_TEST_VENV) && pip install tensorflow-text==2.10.0
38+
@. $(ACTIVATE_TEST_VENV) && pip install tensorflow-text==2.11.0
3939

4040
@echo "Installing test dependencies..."
4141
@. $(ACTIVATE_TEST_VENV) && pip install -r $(CURDIR)/tests/requirements-test.txt
@@ -46,7 +46,7 @@ tlt_notebook_venv: $(CURDIR)/notebooks/requirements.txt
4646

4747
@echo "Installing TF & PYT notebook dependencies..."
4848
@. $(ACTIVATE_NOTEBOOK_VENV) && pip install -r $(CURDIR)/notebooks/requirements.txt && \
49-
pip install tensorflow-text==2.10.0
49+
pip install tensorflow-text==2.11.0
5050

5151
test: unittest integration
5252

@@ -60,7 +60,7 @@ integration: tlt_test_venv
6060

6161
lint: tlt_test_venv
6262
@echo "Style checks..."
63-
@. $(ACTIVATE_TEST_VENV) && flake8 tlt tests
63+
@. $(ACTIVATE_TEST_VENV) && flake8 tlt tests downloader
6464

6565
clean:
6666
rm -rf tlt_test_venv

‎Models.md‎

Lines changed: 187 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,187 @@
1+
# Intel® Transfer Learning Tool Supported Models
2+
3+
## Image Classification
4+
5+
| Model name | Framework | Model Hub |
6+
|------------|-----------|-----------|
7+
| alexnet | PyTorch* | Torchvision* |
8+
| convnext_base | PyTorch | Torchvision |
9+
| convnext_large | PyTorch | Torchvision |
10+
| convnext_small | PyTorch | Torchvision |
11+
| convnext_tiny | PyTorch | Torchvision |
12+
| densenet121 | PyTorch | Torchvision |
13+
| densenet161 | PyTorch | Torchvision |
14+
| densenet169 | PyTorch | Torchvision |
15+
| densenet201 | PyTorch | Torchvision |
16+
| efficientnetv2-b0 | TensorFlow* | TensorFlow Hub* |
17+
| efficientnetv2-b1 | TensorFlow | TensorFlow Hub |
18+
| efficientnetv2-b2 | TensorFlow | TensorFlow Hub |
19+
| efficientnetv2-b3 | TensorFlow | TensorFlow Hub |
20+
| efficientnetv2-s | TensorFlow | TensorFlow Hub |
21+
| efficientnet_b0 | TensorFlow | TensorFlow Hub |
22+
| efficientnet_b0 | PyTorch | Torchvision |
23+
| efficientnet_b1 | TensorFlow | TensorFlow Hub |
24+
| efficientnet_b1 | PyTorch | Torchvision |
25+
| efficientnet_b2 | TensorFlow | TensorFlow Hub |
26+
| efficientnet_b2 | PyTorch | Torchvision |
27+
| efficientnet_b3 | TensorFlow | TensorFlow Hub |
28+
| efficientnet_b3 | PyTorch | Torchvision |
29+
| efficientnet_b4 | TensorFlow | TensorFlow Hub |
30+
| efficientnet_b4 | PyTorch | Torchvision |
31+
| efficientnet_b5 | TensorFlow | TensorFlow Hub |
32+
| efficientnet_b5 | PyTorch | Torchvision |
33+
| efficientnet_b6 | TensorFlow | TensorFlow Hub |
34+
| efficientnet_b6 | PyTorch | Torchvision |
35+
| efficientnet_b7 | TensorFlow | TensorFlow Hub |
36+
| efficientnet_b7 | PyTorch | Torchvision |
37+
| googlenet | PyTorch | Torchvision |
38+
| inception_v3 | TensorFlow | TensorFlow Hub |
39+
| mnasnet0_5 | PyTorch | Torchvision |
40+
| mnasnet1_0 | PyTorch | Torchvision |
41+
| mobilenet_v2 | PyTorch | Torchvision |
42+
| mobilenet_v2_100_224 | TensorFlow | TensorFlow Hub |
43+
| mobilenet_v3_large | PyTorch | Torchvision |
44+
| mobilenet_v3_small | PyTorch | Torchvision |
45+
| nasnet_large | TensorFlow | TensorFlow Hub |
46+
| proxyless_cpu | PyTorch | PyTorch Hub* |
47+
| regnet_x_16gf | PyTorch | Torchvision |
48+
| regnet_x_1_6gf | PyTorch | Torchvision |
49+
| regnet_x_32gf | PyTorch | Torchvision |
50+
| regnet_x_3_2gf | PyTorch | Torchvision |
51+
| regnet_x_400mf | PyTorch | Torchvision |
52+
| regnet_x_800mf | PyTorch | Torchvision |
53+
| regnet_x_8gf | PyTorch | Torchvision |
54+
| regnet_y_16gf | PyTorch | Torchvision |
55+
| regnet_y_1_6gf | PyTorch | Torchvision |
56+
| regnet_y_32gf | PyTorch | Torchvision |
57+
| regnet_y_3_2gf | PyTorch | Torchvision |
58+
| regnet_y_400mf | PyTorch | Torchvision |
59+
| regnet_y_800mf | PyTorch | Torchvision |
60+
| regnet_y_8gf | PyTorch | Torchvision |
61+
| resnet101 | PyTorch | Torchvision |
62+
| resnet152 | PyTorch | Torchvision |
63+
| resnet18 | PyTorch | Torchvision |
64+
| resnet18_ssl | PyTorch | PyTorch Hub |
65+
| resnet18_swsl | PyTorch | PyTorch Hub |
66+
| resnet34 | PyTorch | Torchvision |
67+
| resnet50 | PyTorch | Torchvision |
68+
| resnet50_ssl | PyTorch | PyTorch Hub |
69+
| resnet50_swsl | PyTorch | PyTorch Hub |
70+
| resnet_v1_50 | TensorFlow | TensorFlow Hub |
71+
| resnet_v2_101 | TensorFlow | TensorFlow Hub |
72+
| resnet_v2_50 | TensorFlow | TensorFlow Hub |
73+
| resnext101_32x16d_ssl | PyTorch | PyTorch Hub |
74+
| resnext101_32x16d_swsl | PyTorch | PyTorch Hub |
75+
| resnext101_32x16d_wsl | PyTorch | PyTorch Hub |
76+
| resnext101_32x32d_wsl | PyTorch | PyTorch Hub |
77+
| resnext101_32x48d_wsl | PyTorch | PyTorch Hub |
78+
| resnext101_32x4d_ssl | PyTorch | PyTorch Hub |
79+
| resnext101_32x4d_swsl | PyTorch | PyTorch Hub |
80+
| resnext101_32x8d | PyTorch | Torchvision |
81+
| resnext101_32x8d_ssl | PyTorch | PyTorch Hub |
82+
| resnext101_32x8d_swsl | PyTorch | PyTorch Hub |
83+
| resnext101_32x8d_wsl | PyTorch | PyTorch Hub |
84+
| resnext50_32x4d | PyTorch | Torchvision |
85+
| resnext50_32x4d_ssl | PyTorch | PyTorch Hub |
86+
| resnext50_32x4d_swsl | PyTorch | PyTorch Hub |
87+
| shufflenet_v2_x0_5 | PyTorch | Torchvision |
88+
| shufflenet_v2_x1_0 | PyTorch | Torchvision |
89+
| vgg11 | PyTorch | Torchvision |
90+
| vgg11_bn | PyTorch | Torchvision |
91+
| vgg13 | PyTorch | Torchvision |
92+
| vgg13_bn | PyTorch | Torchvision |
93+
| vgg16 | PyTorch | Torchvision |
94+
| vgg16_bn | PyTorch | Torchvision |
95+
| vgg19 | PyTorch | Torchvision |
96+
| vgg19_bn | PyTorch | Torchvision |
97+
| vit_b_16 | PyTorch | Torchvision |
98+
| vit_b_32 | PyTorch | Torchvision |
99+
| vit_l_16 | PyTorch | Torchvision |
100+
| vit_l_32 | PyTorch | Torchvision |
101+
| wide_resnet101_2 | PyTorch | Torchvision |
102+
| wide_resnet50_2 | PyTorch | Torchvision |
103+
| ConvNeXtBase | TensorFlow | Keras* |
104+
| ConvNeXtLarge | TensorFlow | Keras |
105+
| ConvNeXtSmall | TensorFlow | Keras |
106+
| ConvNeXtTiny | TensorFlow | Keras |
107+
| ConvNeXtXLarge | TensorFlow | Keras |
108+
| DenseNet121 | TensorFlow | Keras |
109+
| DenseNet169 | TensorFlow | Keras |
110+
| DenseNet201 | TensorFlow | Keras |
111+
| EfficientNetB0 | TensorFlow | Keras |
112+
| EfficientNetB1 | TensorFlow | Keras |
113+
| EfficientNetB2 | TensorFlow | Keras |
114+
| EfficientNetB3 | TensorFlow | Keras |
115+
| EfficientNetB4 | TensorFlow | Keras |
116+
| EfficientNetB5 | TensorFlow | Keras |
117+
| EfficientNetB6 | TensorFlow | Keras |
118+
| EfficientNetB7 | TensorFlow | Keras |
119+
| EfficientNetV2B0 | TensorFlow | Keras |
120+
| EfficientNetV2B1 | TensorFlow | Keras |
121+
| EfficientNetV2B2 | TensorFlow | Keras |
122+
| EfficientNetV2B3 | TensorFlow | Keras |
123+
| EfficientNetV2L | TensorFlow | Keras |
124+
| EfficientNetV2M | TensorFlow | Keras |
125+
| EfficientNetV2S | TensorFlow | Keras |
126+
| InceptionResNetV2 | TensorFlow | Keras |
127+
| InceptionV3 | TensorFlow | Keras |
128+
| MobileNet | TensorFlow | Keras |
129+
| MobileNetV2 | TensorFlow | Keras |
130+
| NASNetLarge | TensorFlow | Keras |
131+
| NASNetMobile | TensorFlow | Keras |
132+
| ResNet101 | TensorFlow | Keras |
133+
| ResNet101V2 | TensorFlow | Keras |
134+
| ResNet152 | TensorFlow | Keras |
135+
| ResNet152V2 | TensorFlow | Keras |
136+
| ResNet50 | TensorFlow | Keras |
137+
| ResNet50V2 | TensorFlow | Keras |
138+
| VGG16 | TensorFlow | Keras |
139+
| VGG19 | TensorFlow | Keras |
140+
| Xception | TensorFlow | Keras |
141+
142+
## Text Classification
143+
144+
| Model name | Framework | Model Hub |
145+
|------------|-----------|-----------|
146+
| bert-base-cased | PyTorch | Hugging Face* |
147+
| bert-large-uncased | PyTorch | Hugging Face |
148+
| bert_en_uncased_L-12_H-768_A-12 | TensorFlow | TensorFlow Hub |
149+
| bert_en_wwm_uncased_L-24_H-1024_A-16 | TensorFlow | TensorFlow Hub |
150+
| clinical-bert | PyTorch | Hugging Face |
151+
| distilbert-base-uncased | PyTorch | Hugging Face |
152+
| small_bert/bert_en_uncased_L-10_H-128_A-2 | TensorFlow | TensorFlow Hub |
153+
| small_bert/bert_en_uncased_L-10_H-256_A-4 | TensorFlow | TensorFlow Hub |
154+
| small_bert/bert_en_uncased_L-10_H-512_A-8 | TensorFlow | TensorFlow Hub |
155+
| small_bert/bert_en_uncased_L-10_H-768_A-12 | TensorFlow | TensorFlow Hub |
156+
| small_bert/bert_en_uncased_L-12_H-128_A-2 | TensorFlow | TensorFlow Hub |
157+
| small_bert/bert_en_uncased_L-12_H-256_A-4 | TensorFlow | TensorFlow Hub |
158+
| small_bert/bert_en_uncased_L-12_H-512_A-8 | TensorFlow | TensorFlow Hub |
159+
| small_bert/bert_en_uncased_L-12_H-768_A-12 | TensorFlow | TensorFlow Hub |
160+
| small_bert/bert_en_uncased_L-2_H-128_A-2 | TensorFlow | TensorFlow Hub |
161+
| small_bert/bert_en_uncased_L-2_H-256_A-4 | TensorFlow | TensorFlow Hub |
162+
| small_bert/bert_en_uncased_L-2_H-512_A-8 | TensorFlow | TensorFlow Hub |
163+
| small_bert/bert_en_uncased_L-2_H-768_A-12 | TensorFlow | TensorFlow Hub |
164+
| small_bert/bert_en_uncased_L-4_H-128_A-2 | TensorFlow | TensorFlow Hub |
165+
| small_bert/bert_en_uncased_L-4_H-256_A-4 | TensorFlow | TensorFlow Hub |
166+
| small_bert/bert_en_uncased_L-4_H-512_A-8 | TensorFlow | TensorFlow Hub |
167+
| small_bert/bert_en_uncased_L-4_H-768_A-12 | TensorFlow | TensorFlow Hub |
168+
| small_bert/bert_en_uncased_L-6_H-128_A-2 | TensorFlow | TensorFlow Hub |
169+
| small_bert/bert_en_uncased_L-6_H-256_A-4 | TensorFlow | TensorFlow Hub |
170+
| small_bert/bert_en_uncased_L-6_H-512_A-8 | TensorFlow | TensorFlow Hub |
171+
| small_bert/bert_en_uncased_L-6_H-768_A-12 | TensorFlow | TensorFlow Hub |
172+
| small_bert/bert_en_uncased_L-8_H-128_A-2 | TensorFlow | TensorFlow Hub |
173+
| small_bert/bert_en_uncased_L-8_H-256_A-4 | TensorFlow | TensorFlow Hub |
174+
| small_bert/bert_en_uncased_L-8_H-512_A-8 | TensorFlow | TensorFlow Hub |
175+
| small_bert/bert_en_uncased_L-8_H-768_A-12 | TensorFlow | TensorFlow Hub |
176+
177+
## Image Anomaly Detection
178+
179+
| Model name | Framework | Model Hub |
180+
|------------|-----------|-----------|
181+
| resnet101 | PyTorch | Torchvision |
182+
| resnet152 | PyTorch | Torchvision |
183+
| resnet18 | PyTorch | Torchvision |
184+
| resnet34 | PyTorch | Torchvision |
185+
| resnet50 | PyTorch | Torchvision |
186+
187+
*Other names and brands may be claimed as the property of others. [Trademarks](http://www.intel.com/content/www/us/en/legal/trademarks.html)

0 commit comments

Comments
 (0)