tsts.optimizers

Functions

class tsts.optimizers.build_optimizer[source]

Build optimizer.

Parameters
  • params (Iterable) – Model parameters

  • cfg (CN) – Global configuration

Returns

Built optimizer

Return type

Optimizer

Classes

class tsts.optimizers.Adam[source]

Adam implementation.

Example

OPTIMIZER:
  NAME: "Adam"
  LR: 0.001
class tsts.optimizers.SGD[source]

SGD implementation.

Example

OPTIMIZER:
  NAME: "SGD"
  LR: 0.01
  MOMENTUM: 0.9