Skip to content

v1.7.3

Compare
Choose a tag to compare
@KevinMusgrave KevinMusgrave released this 29 Jan 00:48
· 182 commits to master since this release
f27a8bf

Bug fixes and minor improvements

  • Fixed #472, in which enqueue_idx for CrossBatchMemory could not be passed into DistributedLossWrapper
  • Converted CrossBatchMemory's embedding_memory and label_memory to buffers, so they can be saved and loaded as state dicts and transferred to devices use .to(device).