-
Notifications
You must be signed in to change notification settings - Fork 15
Open
Description
I'm getting an error in this line
| attns['image'] = attn.cpu().detach().numpy() |
None -- @alasdairtran do you have any insight as to why this error might be occurring?
For context I am using my own custom dataset.
Thanks!
attns['image'] = attn.cpu().detach().numpy()
Metadata
Metadata
Assignees
Labels
No labels