I’m not sure I understand your question.
You have created this module to combine a linear and Dropout layer, both with fixed parameters.
It looks like your model just consists of this DropoutFC layer. Usually you put Dropout between two consecutive linear layers. Currently you would drop some of your outputs.
What do you mean by “original python file” ?
Where would you like to put this code?
What class is Train()? It looks like a high-level wrapper.