diff options
author | Anthony Wang | 2022-01-11 12:24:42 -0600 |
---|---|---|
committer | Anthony Wang | 2022-01-11 12:24:42 -0600 |
commit | 3c47bb63daa20dc2fa6de6c0fbc7f6f612fcb0da (patch) | |
tree | 00fb7a92e3c1fffc0774150ba6c58dd23371fee4 /mnist/src/main.rs | |
parent | 6be866b0b8b5ec676159eaa56c09371c02cbd7e1 (diff) |
Diffstat (limited to 'mnist/src/main.rs')
-rw-r--r-- | mnist/src/main.rs | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/mnist/src/main.rs b/mnist/src/main.rs new file mode 100644 index 0000000..88e0483 --- /dev/null +++ b/mnist/src/main.rs @@ -0,0 +1,25 @@ +/* Some very simple models trained on the MNIST dataset. + The 4 following dataset files can be downloaded from http://yann.lecun.com/exdb/mnist/ + These files should be extracted in the 'data' directory. + train-images-idx3-ubyte.gz + train-labels-idx1-ubyte.gz + t10k-images-idx3-ubyte.gz + t10k-labels-idx1-ubyte.gz +*/ + +use anyhow::Result; + +mod mnist_conv; +mod mnist_linear; +mod mnist_nn; + +fn main() -> Result<()> { + let args: Vec<String> = std::env::args().collect(); + let model = if args.len() < 2 { None } else { Some(args[1].as_str()) }; + match model { + None => mnist_nn::run(), + Some("linear") => mnist_linear::run(), + Some("conv") => mnist_conv::run(), + Some(_) => mnist_nn::run(), + } +} |