I tried to run Unbalance with 512x512 and I got an error message.
it was fine with 320x320
what did I do wrong?
Code: Select all
===================================================
12/07/2022 16:07:33 INFO Starting
12/07/2022 16:07:33 INFO ===================================================
12/07/2022 16:07:34 INFO Loading data, this may take a while...
12/07/2022 16:07:34 INFO Loading Model from Unbalanced plugin...
12/07/2022 16:07:34 INFO No existing state file found. Generating.
12/07/2022 16:07:34 INFO Mixed precision compatibility check (mixed_float16): OK\nYour GPU will likely run quickly with dtype policy mixed_float16 as it has compute capability of at least 7.0. Your GPU: NVIDIA GeForce GTX 1660 Ti, compute capability 7.5
12/07/2022 16:07:34 INFO Enabling Mixed Precision Training.
12/07/2022 16:07:35 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 3, 144)
12/07/2022 16:07:35 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 144, 288)
12/07/2022 16:07:35 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 288, 576)
12/07/2022 16:07:36 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 576, 864)
12/07/2022 16:07:38 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 864, 1152)
12/07/2022 16:07:43 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 512, 1600)
12/07/2022 16:07:46 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 400, 1600)
12/07/2022 16:07:49 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 400, 800)
12/07/2022 16:07:50 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 200, 400)
12/07/2022 16:07:51 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 100, 3)
12/07/2022 16:07:51 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 512, 1600)
12/07/2022 16:07:53 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 400, 1600)
12/07/2022 16:07:54 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 400, 800)
12/07/2022 16:07:55 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 200, 400)
12/07/2022 16:07:55 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 100, 1)
12/07/2022 16:07:55 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 560, 2048)
12/07/2022 16:08:00 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 512, 512)
12/07/2022 16:08:01 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 512, 512)
12/07/2022 16:08:02 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 512, 2048)
12/07/2022 16:08:06 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 512, 512)
12/07/2022 16:08:07 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 512, 512)
12/07/2022 16:08:07 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 512, 1024)
12/07/2022 16:08:10 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 256, 256)
12/07/2022 16:08:10 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 256, 256)
12/07/2022 16:08:10 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 256, 512)
12/07/2022 16:08:11 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 128, 3)
12/07/2022 16:08:11 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 560, 2048)
12/07/2022 16:08:14 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 512, 2048)
12/07/2022 16:08:16 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 512, 1024)
12/07/2022 16:08:18 INFO Calculating Convolution Aware Initializer for shape: (3, 3, 256, 512)
12/07/2022 16:08:18 INFO Calculating Convolution Aware Initializer for shape: (5, 5, 128, 1)
12/07/2022 16:08:18 WARNING Model was constructed with shape (None, 16, 16, 560) for input KerasTensor(type_spec=TensorSpec(shape=(None, 16, 16, 560), dtype=tf.float32, name='input_3'), name='input_3', description="created by layer 'input_3'"), but it was called on an input with incompatible shape (None, 16, 16, 512).
12/07/2022 16:08:19 CRITICAL Error caught! Exiting...
12/07/2022 16:08:19 ERROR Caught exception in thread: '_training'
12/07/2022 16:08:23 ERROR Got Exception on main handler: