search

fredzzhang / pvic

[ICCV'23] Official PyTorch implementation for paper "Exploring Predicate Visual Context in Detecting Human-Object Interactions"
BSD 3-Clause "New" or "Revised" License
67 stars 8 forks source link

training error for hicodet #59

Closed hurrrrrrtui closed 3 weeks ago

hurrrrrrtui commented 1 month ago

Dear sir, I am a newcomer to this field. I want to train the HICO dataset on a 4070ti GPU, where the batch size is set to 16 and detector='base'. But the result is an error from Nan. The same code did not report any errors when training the VCOCO dataset. How should I make changes?

hurrrrrrtui commented 1 month ago

@fredzzhang

fredzzhang commented 1 month ago

Hi @hurrrrrrtui,

Please refer to this post for details.

Fred.