Hi,
During the training, we use
model.train() 
for i, ( imgs, labels) in enumerate(train_loader):
    prd = model(imgs)

and
model.eval() 
for i, ( imgs, labels) in enumerate(val_loader):
    prd = model(imgs)

to swtich on/off some functions.
How could I get the train/eval flag in side the model? Is there a build-in parameters that I could use? Or I have to pass the flg like
prd = model(imgs, flg_trn=True)