search

Hzzone / MTLFace

When Age-Invariant Face Recognition Meets Face Age Synthesis: A Multi-Task Learning Framework (CVPR 2021 oral & TPAMI 2022)
170 stars 30 forks source link

Get the weight after training #6

Closed ngthalogit closed 2 years ago

ngthalogit commented 2 years ago

Thanks for your great job! After train AIFR/FAS, how can i save the model ? Do we use checkpoint function in ops.py file ?

Hzzone commented 2 years ago

Call the function https://github.com/Hzzone/MTLFace/blob/4ed43c77a5b671fb609f3e010845a2a423be3865/common/ops.py#L100