search

landskape-ai / triplet-attention

Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]
https://openaccess.thecvf.com/content/WACV2021/html/Misra_Rotate_to_Attend_Convolutional_Triplet_Attention_Module_WACV_2021_paper.html
MIT License
406 stars 49 forks source link

wandb error #10

Closed haoyao0131 closed 3 years ago

haoyao0131 commented 3 years ago

Hi, I came across the following error. Does anyone has similar experience? Thanks.

image
iyaja commented 3 years ago

I haven't seen this error before, but I don't think it's related to triplet attention. Is this something that works fine without triplet attention? Are you using one of our training scripts, or did you make modifications?

My guess is that the issue might be related to wandb.watch(model) if you're using that function. Maybe try removing that if you can.

haoyao0131 commented 3 years ago

Thanks for your reply. I modified the code to apply my own dataset. And it works when I remove wandb function.

digantamisra98 commented 3 years ago

Great, thanks for notifying us.

AdarshSantoria commented 6 months ago

@haoyao0131 Do you have fine tuning code without wandb?