-
Notifications
You must be signed in to change notification settings - Fork 13
Open
Description
Line 36 in f082a57
| classes = F.sigmoid(feature) |
classes = F.sigmoid(feature) #feature shape=[1,1], classes shape=[1,1]
one_hot, _ = classes.max(dim=-1) # one_hot shape=[1]
self.model.zero_grad()
one_hot.backward()
This one_hot is not the one hot label, I don't know it is correct or not,
In my experiment, there is a class without any activation hot map. Did you know why? Thanks
Metadata
Metadata
Assignees
Labels
No labels