I want to train a DBN model, but I'm facing this error
Classifier
Train Step: 0Traceback (most recent call last):
File "../../trainer.py", line 60, in <module>
main()
File "../../trainer.py", line 54, in main
model.Train()
File "/home/aries/Documents/deepnet-master/deepnet/neuralnet.py", line 632, in Train
losses = self.TrainOneBatch(step)
File "/home/aries/Documents/deepnet-master/deepnet/neuralnet.py", line 330, in TrainOneBatch
losses2 = self.BackwardPropagate(step)
File "/home/aries/Documents/deepnet-master/deepnet/neuralnet.py", line 316, in BackwardPropagate
loss = self.ComputeDown(node, step)
File "/home/aries/Documents/deepnet-master/deepnet/neuralnet.py", line 218, in ComputeDown
loss = layer.GetLoss(get_deriv=True)
File "/home/aries/Documents/deepnet-master/deepnet/softmax_layer.py", line 64, in GetLoss
state.get_softmax_correct(data, target=temp)
File "/home/aries/Documents/deepnet-master/cudamat/cudamat.py", line 1290, in get_softmax_correct
assert target.shape == labels.shape
AssertionError
Can anyone tell, anything wrong with the model?
FYI, This model is using output node more than 1 node to achieve the multi-label classification problem.
I want to train a DBN model, but I'm facing this error
Can anyone tell, anything wrong with the model?
FYI, This model is using output node more than 1 node to achieve the multi-label classification problem.
thank you