I want to know when I pass model.cuda() or model.train() in paramters of function, do I have to re-assign cuda() or train() ?
def test():
     model = model.cuda()
     testB(model)

def testB(model)
     Should I model.cuda() in here?
     model = model.train()
     testC(model)

def testC(model)
     In here, we have to assign model = model.train() or model.cuda() ?