Skip to content
Snippets Groups Projects
Commit b50bc48c authored by tomrink's avatar tomrink
Browse files

fix a maybe bug

parent 377cac8a
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
modules/deeplearning/espcn.py
+ 1
− 1
View file @ b50bc48c
...@@ -689,7 +689,7 @@ class ESPCN: ...@@ -689,7 +689,7 @@ class ESPCN:
ckpt.restore(ckpt_manager.latest_checkpoint) ckpt.restore(ckpt_manager.latest_checkpoint)
data = normalize(nda_lr, data_param, mean_std_dct) data = normalize(nda_lr, param, mean_std_dct)
data = np.expand_dims(data, axis=0) data = np.expand_dims(data, axis=0)
data = np.expand_dims(data, axis=3) data = np.expand_dims(data, axis=3)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment