Files
2026-07-13 12:45:29 +08:00

26 lines
595 B
TOML

[package]
authors = ["laggui <lagrange.guillaume.1@gmail.com>"]
edition.workspace = true
license.workspace = true
name = "custom-image-dataset"
description = "Example implementation for loading a custom image dataset from disk"
publish = false
version.workspace = true
[lints]
workspace = true
[features]
default = ["burn/std", "burn/tui"]
tch-gpu = ["burn/tch"]
wgpu = ["burn/wgpu"]
metal = ["burn/metal"]
vulkan = ["burn/vulkan"]
[dependencies]
burn = { workspace = true, features = ["train", "vision", "network"] }
# File download
flate2 = { workspace = true }
tar = { workspace = true }