search

naver-ai / egtr

[CVPR 2024 Best paper award candidate] EGTR: Extracting Graph from Transformer for Scene Graph Generation
https://arxiv.org/abs/2404.02072
Apache License 2.0
78 stars 2 forks source link

validation_epoch_end(self, outputs): #12

Open ha1ha2hahaha opened 3 weeks ago

ha1ha2hahaha commented 3 weeks ago

this method not have the “outputs”。。。。how to fix it image

jinbae commented 2 weeks ago

https://github.com/Lightning-AI/pytorch-lightning/issues/19419

Please check your version. https://github.com/naver-ai/egtr/blob/7f87450f32758ed8583948847a8186f2ee8b21e3/train_egtr.py#L339

Our code uses validation_epoch_end instead of on_validation_epoch_end.