Skip to content

Beta 2 Release

Pre-release
Pre-release
Compare
Choose a tag to compare
@cedrickchee cedrickchee released this 12 Nov 16:24
· 40 commits to master since this release
v0.1.0
49ff8a0

Added

  • Implemented reconstruction loss.
  • Saving reconstructed image as file.
  • Improve training speed by using PyTorch DataParallel to wrap our model.
    • PyTorch will parallelized the model and data over multiple GPUs.
  • Supports training:
    • on CPU (tested with macOS Sierra)
    • on one GPU (tested with 1 Tesla K80 GPU)
    • on multiple GPU (tested with 8 GPUs)
    • with or without CUDA (tested with CUDA version 8.0.61)
    • cuDNN 5 (tested with cuDNN 5.1.3)

Changed

  • More intuitive variable naming.

Fixed

  • Resolve Pylint warnings and reformat code.
  • Missing square in equation 4 for margin (class) loss.
Assets 2
Loading