Sorry, ptrblck. Just now I didn’t catch what you meant.
‘Would you like to add DropoutFC somewhere into Train?’  - what did you mean?
I have added DropoutFC on the top as shown above.
You meant I also should add this kind of codes somewhere else in the Train?