Skip to content
simplegrad
.set_train_mode()
Initializing search
simplegrad/simplegrad
simplegrad
simplegrad/simplegrad
Home
Introduction
Core
Core
Tensor & Autograd
Function & Context
Compound Ops
Dtypes
Factory Functions
Module
Optimizer
Scheduler
Functions
Functions
Math
Activations
Reductions
Transform
Conv2d
Pooling
Losses
NN Layers
NN Layers
Linear
Conv2d
Sequential
Dropout
MaxPool2d
Flatten
Embedding
Activations
Losses
Optimizers
Optimizers
SGD
Adam
Schedulers
Schedulers
Overview
Tracking
Tracking
Overview
Visualization
Visualization
Inline
SimpleBoard
Table of contents
set_train_mode
.set_train_mode()
Switch this module and all sub-modules to training mode.