DesignCreator should inherit from torch.nn.Module - Githubissues

es-ude / elastic-ai.creator

elastic ai.creator

MIT License

16 stars 2 forks source link

DesignCreator should inherit from torch.nn.Module #376

Closed julianhoever closed 2 months ago

julianhoever commented 2 months ago

feat: abstract class DesignCreator inherits from torch.nn.Module
refactor: move design_creator module to nn package
refactor: rename the module design_creator to design_creator_module