Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

tf.keras.layers.UpSampling2D crashes(aborts) when size is large #46914

Closed
DNXie opened this issue Feb 4, 2021 · 4 comments · Fixed by #51497
Closed

tf.keras.layers.UpSampling2D crashes(aborts) when size is large #46914

DNXie opened this issue Feb 4, 2021 · 4 comments · Fixed by #51497
Assignees
Labels
comp:keras Keras related issues stat:awaiting tensorflower Status - Awaiting response from tensorflower TF 2.4 for issues related to TF 2.4 type:bug Bug

Comments

@DNXie
Copy link

DNXie commented Feb 4, 2021

System information

  • Have I written custom code (as opposed to using a stock example script provided in TensorFlow): No
  • OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Linux Ubuntu 18.04
  • Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on mobile device: N/A
  • TensorFlow installed from (source or binary): binary
  • TensorFlow version (use command below):2.1.0
  • Python version:3.7.6
  • Bazel version (if compiling from source):N/A
  • GCC/Compiler version (if compiling from source):N/A
  • CUDA/cuDNN version:N/A
  • GPU model and memory:N/A

Describe the current behavior
tf.keras.layers.UpSampling2D crashes(aborts) when size is large
Describe the expected behavior
expect an exception message if the input unexpected instead of crash.

Standalone code to reproduce the issue

import tensorflow as tf
import numpy as np
tf.keras.layers.UpSampling2D(size=1610637938, data_format='channels_first', interpolation='bilinear')(np.ones((5,1,1,1)))

Output:

2021-02-04 04:44:48.936606: F tensorflow/core/framework/tensor_shape.cc:353] Check failed: 0 <= new_num_elements (0 vs. -5475971237085092396)
Aborted (core dumped)
@amahendrakar
Copy link
Contributor

Was able to reproduce the issue with TF v2.3, TF v2.4 and TF-nightly. Please find the gist of it here. Thanks!

@amahendrakar amahendrakar added comp:keras Keras related issues TF 2.4 for issues related to TF 2.4 labels Feb 4, 2021
@amahendrakar amahendrakar assigned ymodak and unassigned amahendrakar Feb 4, 2021
@ymodak ymodak added the stat:awaiting tensorflower Status - Awaiting response from tensorflower label Feb 4, 2021
@saikumarchalla
Copy link

Colab crashes till in TF 2.6 Nightly as well. Please find the gist here.Thanks!

@yongtang
Copy link
Member

Added a PR #51497 for the fix.

@google-ml-butler
Copy link

Are you satisfied with the resolution of your issue?
Yes
No

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
comp:keras Keras related issues stat:awaiting tensorflower Status - Awaiting response from tensorflower TF 2.4 for issues related to TF 2.4 type:bug Bug
Projects
None yet
Development

Successfully merging a pull request may close this issue.

5 participants