Skip to content

0.0.3

Compare
Choose a tag to compare
@anjandeepsahni anjandeepsahni released this 22 Jan 13:52
· 28 commits to master since this release

New features:

  • FLOPs support for softmax, embedding, sigmoid, tanh, lstm, rnn, gru.
  • Calculate and track running statistics of data.
  • Get statistics of dataset (mean, standard deviation and variance).
  • Calculate and track accuracy of predictions.
  • Calculate and track hamming loss of predictions.
  • Track and maintain running average of loss.
  • Plot (average) gradients for each layer in model.

Bug fixes:

  • Multi-input model support for model summary and FLOPs.

Deprecations:

  • get_current_lr() is deprecated, use get_lr() instead.
  • set_current_lr() is deprecated, use set_lr() instead.