Skip to content

Commit 91f445e

Browse files
committed
Update weight conversion for new layers, and new data format
1 parent bb56bb9 commit 91f445e

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

weight_conversion_theano.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def shuffle_rows(original_w, nb_last_conv, nb_rows_dense):
7979
first_dense = True
8080
nb_last_conv = 0
8181

82-
for dirpath in ["tf-kernels-tf-dim-ordering/", "tf-kernels-th-dim-ordering/", "th-kernels-tf-dim-ordering/"]:
82+
for dirpath in ["tf-kernels-channels-last-dim-ordering/", "tf-kernels-channels-first-dim-ordering/", "th-kernels-channels-last-dim-ordering/"]:
8383
if not os.path.exists(dirpath):
8484
os.makedirs(dirpath)
8585

weight_convertion_tensorflow.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def shuffle_rows(original_w, nb_last_conv, nb_rows_dense):
7979
first_dense = True
8080
nb_last_conv = 0
8181

82-
for dirpath in ["th-kernels-th-dim-ordering/", "tf-kernels-th-dim-ordering/", "th-kernels-tf-dim-ordering/"]:
82+
for dirpath in ["th-kernels-channels-first-dim-ordering/", "tf-kernels-channels-first-dim-ordering/", "th-kernels-channels-last-dim-ordering/"]:
8383
if not os.path.exists(dirpath):
8484
os.makedirs(dirpath)
8585

0 commit comments

Comments
 (0)