burn-train 0.21.0-pre.5

Training crate for the Burn framework
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
mod dice;
mod dists;
mod fid;
mod lpips;
mod ms_ssim;
mod psnr;
mod ssim;

pub use dice::*;
pub use dists::*;
pub use fid::*;
pub use lpips::*;
pub use ms_ssim::*;
pub use psnr::*;
pub use ssim::*;