InfoNCE loss support and Resume training support

facebookresearch / EgoVLPv2

Code release for "EgoVLPv2: Egocentric Video-Language Pre-training with Fusion in the Backbone" [ICCV, 2023]

MIT License

85 stars 11 forks source link

InfoNCE loss support and Resume training support #6

Closed ShramanPramanick closed 9 months ago

ShramanPramanick commented 9 months ago

The codebase now supports InfoNCE loss by setting "verb" and "noun" to false in egoclip.json, and supports resume training from existing checkpoint using args.resume.