pystiche / papers

Reference implementation and replication of prominent NST papers
BSD 3-Clause "New" or "Revised" License
4 stars 1 forks source link

Add a Config class for the different parameter setting sets #297

Open jbueltemeier opened 1 year ago

jbueltemeier commented 1 year ago

As mentioned in #296 it is currently difficult to deal with various parameter settings such as impl_params: bool, instance_norm: bool as well as parameters provided by the author.

In the future we should change these parameters in favour of config: GatysConfig, like

class GatysConfig(enum.Enum):
    PAPER = enum.auto()
    REFERENCE_IMPLEMENTATION = enum.auto()
    AUTHOR = enum.auto()