Code

Logits#67

Code

Open

thatguy11325 wants to merge 4 commits intoPufferAI:0.7PufferAI/PufferLib:0.7from thatguy11325:logitsthatguy11325/PufferLib:logitsCopy head branch name to clipboard

+52-31

Commits

Commits on Jan 28, 2024

• Add torch.multinomial-based logit sampler for speedups

  thatguy11325

  committed
  fac330c

  Copy full SHA for fac330c

• Dont shadow function name

  thatguy11325

  committed
  e1bc330

  Copy full SHA for e1bc330

• Add some typing. Normalized logits should be a list

  thatguy11325

  committed
  61f8c9c

  Copy full SHA for 61f8c9c

• fix cleanrl issue. Add a squeeze

  thatguy11325

  committed
  9966e95

  Copy full SHA for 9966e95