krisrs1128 / clouds_dist

Simulation of low-clouds, from weather measures.
4 stars 1 forks source link

Allow partial loading from checkpoints #143

Closed krisrs1128 closed 4 years ago

krisrs1128 commented 4 years ago

This closes #115. It now allows you to partially load a checkpoint, based on regexes describing parameter names. For example, in the new default.yaml, it will initialize the down layers in the U-Net generator, and all parameters of the discriminator (of course, it won't execute unless init_chkpt_dir is set to an actual path, which isn't specified in the default).

This requires adding an argument to resume(), but the default for this argument matches the previous behavior.