huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 03:54:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use byteorder in mnist. (#785)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-09 09:03:59 +01:00
committed by GitHub
parent 976a1086ee
commit 722c50bb0c
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-datasets/src/vision/mnist.rs
View File

@ -8,10 +8,9 @@ use parquet::file::reader::{FileReader, SerializedFileReader};
use std::fs::File; use std::fs::File;
use std::io::{self, BufReader, Read}; use std::io::{self, BufReader, Read};
fn read_u32<T: Read>(reader: &mut T) -> Result<u32> { fn read_u32<T: Read>(reader: &mut T) -> std::io::Result<u32> {
let mut b = [0u8; 4]; use byteorder::ReadBytesExt;
reader.read_exact(&mut b)?; reader.read_u32::<byteorder::BigEndian>()
Ok(u32::from_be_bytes(b))
} }
fn check_magic_number<T: Read>(reader: &mut T, expected: u32) -> Result<()> { fn check_magic_number<T: Read>(reader: &mut T, expected: u32) -> Result<()> {