You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
thank you
The text was updated successfully, but these errors were encountered:
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
The text was updated successfully, but these errors were encountered: