grey-eye / talking-heads

Our implementation of "Few-Shot Adversarial Learning of Realistic Neural Talking Head Models" (Egor Zakharov et al.)
GNU General Public License v3.0
593 stars 110 forks source link

Update run.py to save some memory #7

Closed ShuangLiu1992 closed 5 years ago

ShuangLiu1992 commented 5 years ago

avoid using retain_graph=True to save memory