crowdcount-cascaded-mtl icon indicating copy to clipboard operation
crowdcount-cascaded-mtl copied to clipboard

Published 20 hours ago • verified publisher icon svishwa

Softmax runtime warning

Open hiramustafa77 opened this issue 3 years ago • 0 comments

UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument. this is the original code line: density_cls_prob = F.softmax(density_cls_score) So how do I know whether dim should be taken '0' or '1'

hiramustafa77 avatar Oct 21 '21 08:10 hiramustafa77