VTGAN icon indicating copy to clipboard operation
VTGAN copied to clipboard

train problem: Incompatible shapes: [2,64,64,1] vs. [2,32,32,1]

Open zhangbaijin opened this issue 2 years ago • 2 comments

I am sorry to ask many questions, when i run train.py,there is a problem, and my tf==2.5.0 , keras==2.5.0, appreciate it, best wishes :D Traceback (most recent call last): File "train.py", line 220, in <module> train(d_model1, d_model2, g_model_coarse, g_model_fine, vt_model, dataset, n_epochs=args.epochs, File "train.py", line 73, in train d_loss4 = d_model2.train_on_batch([X_realA_half,X_fakeB_half],[y1_coarse,y2,d_feat2_fake[2]])[0] # [,X_fakeB_half] File "/opt/conda/lib/python3.8/site-packages/keras/engine/training.py", line 1800, in train_on_batch logs = self.train_function(iterator) File "/opt/conda/lib/python3.8/site-packages/keras/engine/training.py", line 830, in train_function return step_function(self, iterator) File "/opt/conda/lib/python3.8/site-packages/keras/engine/training.py", line 820, in step_function outputs = model.distribute_strategy.run(run_step, args=(data,)) File "/opt/conda/lib/python3.8/site-packages/tensorflow/python/distribute/distribute_lib.py", line 1285, in run return self._extended.call_for_each_replica(fn, args=args, kwargs=kwargs) File "/opt/conda/lib/python3.8/site-packages/tensorflow/python/distribute/distribute_lib.py", line 2833, in call_for_each_replica return self._call_for_each_replica(fn, args, kwargs) File "/opt/conda/lib/python3.8/site-packages/tensorflow/python/distribute/distribute_lib.py", line 3608, in _call_for_each_replica return fn(*args, **kwargs) File "/opt/conda/lib/python3.8/site-packages/tensorflow/python/autograph/impl/api.py", line 597, in wrapper return func(*args, **kwargs) File "/opt/conda/lib/python3.8/site-packages/keras/engine/training.py", line 813, in run_step outputs = model.train_step(data) File "/opt/conda/lib/python3.8/site-packages/keras/engine/training.py", line 771, in train_step loss = self.compiled_loss( File "/opt/conda/lib/python3.8/site-packages/keras/engine/compile_utils.py", line 201, in __call__ loss_value = loss_obj(y_t, y_p, sample_weight=sw) File "/opt/conda/lib/python3.8/site-packages/keras/losses.py", line 142, in __call__ losses = call_fn(y_true, y_pred) File "/opt/conda/lib/python3.8/site-packages/keras/losses.py", line 246, in call return ag_fn(y_true, y_pred, **self._fn_kwargs) File "/opt/conda/lib/python3.8/site-packages/tensorflow/python/util/dispatch.py", line 206, in wrapper return target(*args, **kwargs) File "/opt/conda/lib/python3.8/site-packages/keras/losses.py", line 1202, in mean_squared_error return backend.mean(tf.math.squared_difference(y_pred, y_true), axis=-1) File "/opt/conda/lib/python3.8/site-packages/tensorflow/python/ops/gen_math_ops.py", line 10405, in squared_difference _ops.raise_from_not_ok_status(e, name) File "/opt/conda/lib/python3.8/site-packages/tensorflow/python/framework/ops.py", line 6897, in raise_from_not_ok_status six.raise_from(core._status_to_exception(e.code, message), None) File "<string>", line 3, in raise_from tensorflow.python.framework.errors_impl.InvalidArgumentError: Incompatible shapes: [2,64,64,1] vs. [2,32,32,1] [Op:SquaredDifference]

zhangbaijin avatar Mar 02 '22 03:03 zhangbaijin