0.3.1

fix bug

0.3.0

make sure when sampling, batch does not exceed training batch size

0.2.4

allow for mixed precision training with fp16 flag

0.2.3

allow for mixed precision training with fp16 flag

0.2.2

fix loading checkpoint

0.2.1

update ema more frequently

0.2.0

add exponential moving average of model, also allow passing in custom…

0.1.4

fix small bug with gradient accumulation

0.1.3

add interpolation

0.1.2

fix another stray bug