Open ILTShade opened 6 years ago
in the center_loss.py, the shape of ground truth label should be (batch_size), not num_classes and I think it can be good, if you decrease using of cuda() in center_loss.py
in the center_loss.py, the shape of ground truth label should be (batch_size), not num_classes and I think it can be good, if you decrease using of cuda() in center_loss.py