Hey it seems like solving my problem, but do i have to replace every instance of model.“abc” with model.module.abc
like :model.train() to model.module.train()
model.eval() to model.module.eval() and so on
Please let me know it there are some exceptions to this ?