Cargo Features
[dependencies]
burn-core = { version = "0.17.0", default-features = false, features = ["std", "dataset", "network", "sqlite", "sqlite-bundled", "vision", "audio", "doc", "record-item-custom-serde", "experimental-named-tensor", "test-cuda", "test-rocm", "test-tch", "test-wgpu", "test-wgpu-spirv", "burn-remote", "burn-router"] }
- default = std
-
These default features are set whenever
burn-core
is added without
somewhere in the dependency tree.default-features = false Enables burn-common, optional burn-dataset, and burn-tensor
burn-common:
** Please make sure all dependencies support no_std when std is disabled **
- std default doc? = flate2, log, rmp-serde
-
Enables std of bincode, burn-common, burn-tensor, half, num-traits, rand, serde, and serde_json
rand:
Default enables std
Affects
config::Config.save
,config::Config.load
,burn-core::data
,burn-core::lr_scheduler
,base::Module.save_file
,base::Module.load_file
,memory::NamedMpkBytesRecorder
,recorder::DefaultRecorder
,recorder::CompactRecorder
,recorder::SensitiveCompactRecorder
,recorder::NoStdTrainingRecorder
,recorder::DebugRecordSettings
… - dataset doc? = burn-dataset
-
Affects
data::dataloader
,data::dataset
… - network
-
Enables network of burn-common
Affects
data::network
… - sqlite
-
Enables sqlite of optional burn-dataset
- sqlite-bundled
-
Enables sqlite-bundled of optional burn-dataset
- vision doc?
-
Enables network of burn-common, vision of optional burn-dataset
- audio doc?
-
Enables audio of optional burn-dataset
- doc = audio, dataset, std, vision
-
Enables doc of burn-common, burn-dataset, and burn-tensor
- record-item-custom-serde = regex, thiserror
-
Custom deserializer for Record that is helpful for importing data, such as PyTorch pt files.
Affects
record::serde
… - experimental-named-tensor
-
Serialization formats
Enables experimental-named-tensor of burn-tensor
- test-cuda
-
To use cuda during testing, default uses ndarray.
Enables burn-cuda
FOR TESTING
- test-rocm
-
To use hip during testing, default uses ndarray.
Enables burn-rocm
- test-tch
-
To use tch during testing, default uses ndarray.
Enables burn-tch
- test-wgpu
-
To use wgpu during testing, default uses ndarray.
Enables burn-wgpu
- test-wgpu-spirv
-
To use wgpu-spirv during testing, default uses ndarray.
Features from optional dependencies
In crates that don't use the dep:
syntax, optional dependencies automatically become Cargo features. These features may have been created by mistake, and this functionality may be removed in the future.
- burn-dataset dataset? doc?
- log std
- flate2 std
-
Enables flate2
Serialize Deserialize
- regex record-item-custom-serde?
- rmp-serde std
- thiserror record-item-custom-serde?
- burn-cuda test-cuda?
- burn-rocm test-rocm?
- burn-remote implicit feature
-
Enables burn-remote
burn-remote:
Backend router decorator over websocket
- burn-router implicit feature
-
Enables burn-router
burn-router:
Multi-backend router decorator for the Burn framework
- burn-tch test-tch?
- burn-wgpu test-wgpu? test-wgpu-spirv?