lamp-core/lamp.nn/TrainingMode

TrainingMode

trait TrainingMode[M]

Type class about how to switch a module into training or evaluation mode

Companion:
object
class Object
trait Matchable
class Any

Value members

Abstract methods

def asEval(m: M): M
def asTraining(m: M): M