NVIDIA Deep Learning Examples
SIM for TensorFlow2
Resource
NVIDIA Deep Learning Examples
SIM for TensorFlow2

Search-based Interest Model (SIM) is a system for predicting user behavior given sequences of previous interactions.

Changelog

May 2022

  • Initial release

Known issues

  • While benchmarking inference on a single GPU, sometimes throughput drops drastically in the middle of the epoch and remains low until the end of the epoch.
  • On a multi-GPU setup, the summary of throughput (in the last line of the logfile) is lower than it would result from each step`s throughput (sample/s). It is probably the case when a single GPU is slower than the one on the logging node. In this case, the overhead for synchronization before the final throughput calculation is higher than usual.
  • The SIM model results are non-deterministic, even using the same random seed. The reason for this non-determinism is the tf.math.unsorted_segment_sum operation called within an optimization step. Its influence depends on categorical data distribution within a batch, and this issue is more severe for momentum-based optimizers. A potential solution is to use a deterministic version of this op which allows perfect reproduction, but is up to six times slower training.