Skip to content

Bug fix

Pre-release
Pre-release
Compare
Choose a tag to compare
@w86763777 w86763777 released this 09 Jul 09:01
· 35 commits to master since this release
  • Fix CUDA is reinitialized in subprocess when multiprocess is used for distributed training.
  • Add argument device with default value torch.device('cuda:0') to core.get_inception_feature which can be assigned by **kwargs for all utility functions utils.*.