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.
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.