Open
Description
Getting an error when running model.fit in the multiple category training example.
history = model.fit( train_dataloader, steps_per_epoch=len(train_dataloader), epochs=EPOCHS, callbacks=callbacks, validation_data=valid_dataloader, validation_steps=len(valid_dataloader), )
Any ideas on what may be causing it? I suspect it has to do with my train_dataloader object, but I've prepared it as shown in the example.
ValueError: in user code:
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\keras\engine\training.py:784 train_function *
return step_function(self, iterator)
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\keras\engine\training.py:774 step_function **
outputs = model.distribute_strategy.run(run_step, args=(data,))
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\distribute\distribute_lib.py:1261 run
return self._extended.call_for_each_replica(fn, args=args, kwargs=kwargs)
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\distribute\distribute_lib.py:2794 call_for_each_replica
return self._call_for_each_replica(fn, args, kwargs)
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\distribute\distribute_lib.py:3217 _call_for_each_replica
return fn(*args, **kwargs)
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\keras\engine\training.py:767 run_step **
outputs = model.train_step(data)
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\keras\engine\training.py:733 train_step
y_pred = self(x, training=True)
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\keras\engine\base_layer.py:977 __call__
input_spec.assert_input_compatibility(self.input_spec, inputs, self.name)
F:\anaconda3\envs\tf-n-gpu\lib\site-packages\tensorflow\python\keras\engine\input_spec.py:204 assert_input_compatibility
raise ValueError('Layer ' + layer_name + ' expects ' +
ValueError: Layer model_8 expects 1 input(s), but it received 2 input tensors. Inputs received: [<tf.Tensor 'IteratorGetNext:0' shape=(None, None, None, None) dtype=uint8>, <tf.Tensor 'IteratorGetNext:1' shape=(None, None, None, None) dtype=float32>]
Metadata
Metadata
Assignees
Labels
No labels