danielward27 / flowjax

https://danielward27.github.io/flowjax/
MIT License
85 stars 10 forks source link

train_flow update #34

Closed danielward27 closed 1 year ago

danielward27 commented 1 year ago

Minor changes to the training script, allowing val_prop=0, and avoiding dropping the last batch if it is the same shape as previous batches (not truncated).