Is Module.eval() same as Module.train(False)?

[RahulBhalley]

In documentation, we have Module.eval() and Module.train(mode). Is setting mode to False same as using Module.eval()?

[angeloskath]

Yes it is literally how Module.eval() is implemented.

mlx/python/mlx/nn/layers/base.py

Line 543 in f30e633

def eval(self):

[RahulBhalley]

Ok, thanks. :)