Reproduce Discrete Soft Actor Critic in TF 2.0
junhuang-ifast opened this issue · 1 comments
junhuang-ifast commented
Hi are there any pointers on how to reproduce the Discrete SAC code in tf2? Especially the torch.gather() which does not particularly behave the same way as tf.gather or tf.gather_nd. Any help would be appreciated !
junhuang-ifast commented
solved this using a hacky way with tf.gather_nd