Skip to content

Commit 09f0262

Browse files
authored
Add TF2 versions of TF sample models (#225)
* move tf sample models into tf1 and tf2 folders * add tf2 versions of sample models * remove unused file * update README * update yolov3
1 parent c3f16a7 commit 09f0262

File tree

128 files changed

+2670
-38
lines changed

Some content is hidden

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

128 files changed

+2670
-38
lines changed

TensorFlow/.gitignore

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1-
/squeezenet/data
1+
TF1/squeezenet/data
22
.vscode
33
__pycache__
44
traces/
55
!/op_report/data/traces
6-
*.xlsx
6+
*.xlsx
7+
saved_models/

TensorFlow/README.md

Lines changed: 12 additions & 3 deletions

TensorFlow/VisionTransformer/train_cifar10.py renamed to TensorFlow/TF1/VisionTransformer/train_cifar10.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def get_train_dataset(train_dataset):
8282
],
8383
)
8484

85-
file_path= './saved_models/Model_Cifar100'
85+
file_path= './saved_models/Model_Cifar10'
8686
checkpoint = ModelCheckpoint(file_path, monitor='val_Top-1-accuracy', verbose=1, save_best_only=True, mode='max')
8787
reduce_on_plateau = ReduceLROnPlateau(monitor="val_Top-1-accuracy", mode="max", factor=0.5, patience=PATIENCE, verbose=1,min_lr=0.00002)
8888
callbacks_list = [checkpoint, reduce_on_plateau]

0 commit comments

Comments
 (0)