v0.0.1.dev4

cfg

• renamed loss_rediction mode batch_mean to mean_sum
• changed default loss_reduction mode to mean
• changed default beta to match new loss_reduction mode

flatness metric

• added average angle along traversals
• simplified greatly

reconstruction loss

• more reconstruction losses based off of distributions
  • bernoulli
  • continuous_bernoulli
  • normal

shedules

• new schedules:
  • NoopSchedule: does absolutely nothing!
  • CosineWaveSchedule: smooth cosine wave
• Adjusted arguments of most schedules

bugs

• fixed various runtime bugs
  • logging crash for W&B
  • disabled checkpointing in trainer
  • flatness metric did not support axis size < 2