26 lines
595 B
TOML
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 }
|