I created a class (named DividedLinear) which is a modified version of existing Linear class and tested the new class and it’s autograd in a model, it all worked.
Then I added the new class to the bottom of file: /Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/torch/nn/modules/linear.py
to be able to call torch.nn.DividedLinear in a model’s torch.nn.Sequential. However I get the following error: module ‘torch.nn’ has no attribute ‘DividedLinear’
What other modifications do I need to do or should I just keep the new class separate?
P.s. I am new.