torchscale icon indicating copy to clipboard operation
torchscale copied to clipboard

Config fix

Open agoryuno opened this issue 11 months ago • 0 comments

I've rewritten the torchscale.architecture.config module to use inheritance and remove the redundant code. There are now 3 classes:

Config - that holds all common options EncoderConfig - inherits 'Config' and holds Encoder specific options DecoderConfig - inherits 'Config' and holds Decoder specific options EncoderDecoderConfig - inherits EncoderConfig and DecoderConfig and holds that one option shared by Encoder and Decoder objects.

I also took the liberty to add some type hints for args parameters in encoder/decoder modules.

agoryuno avatar Jul 10 '23 18:07 agoryuno