GAN learner.save() for training in stages

Part 1 (2019)
1

I left a GAN training last night with

learn.save("gan-stage1")
learn.export("gan.pkl")

absent mindedly thrown in at the end. the save() was fine but the export() blew up because something wouldn’t serialize which made me realise i’d never want to export the gan learner anyway.

however, for overnight training in stages what do i need to save?
if i save() the gan learner can i just load that back up and carry on training or should i save() the gen and the critic seperately and and load them back into a new gan learner in the next session?

1 Like