Skip to content

Commit 07c85d1

Browse files
committed
chore: update burn features in examples to include autotune and remove autodiff
1 parent c0ee89f commit 07c85d1

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

examples/resnet-finetune/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ workspace = true
1313
anyhow = { workspace = true }
1414
bimm = { version = "0.19.1", path = "../../crates/bimm" }
1515

16-
burn = { workspace = true, features = ["network", "train", "autodiff", "vision"] }
16+
burn = { workspace = true, features = ["train", "fusion", "autotune", "network", "vision"] }
1717

1818
clap = { workspace = true, features = ["derive"] }
1919

examples/resnet_tiny/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ bimm = { version = "0.19.1", path = "../../crates/bimm" }
1414
bimm-firehose = { version = "0.19.1", path = "../../crates/bimm-firehose" }
1515
bimm-firehose-image = { version = "0.19.1", path = "../../crates/bimm-firehose-image" }
1616

17-
burn = { workspace = true, features = ["train", "autodiff", "fusion", "vision", "dataset"] }
17+
burn = { workspace = true, features = ["train", "fusion", "autotune", "vision", "dataset"] }
1818
serde = { workspace = true, features = ["derive"] }
1919

2020
clap = { workspace = true, features = ["derive"] }

examples/swin_tiny/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ bimm = { version = "0.19.1", path = "../../crates/bimm" }
1414
bimm-firehose = { version = "0.19.1", path = "../../crates/bimm-firehose" }
1515
bimm-firehose-image = { version = "0.19.1", path = "../../crates/bimm-firehose-image" }
1616

17-
burn = { workspace = true, features = ["train", "autodiff", "cuda", "fusion", "vision", "dataset"] }
17+
burn = { workspace = true, features = ["train", "fusion", "autotune", "vision", "dataset"] }
1818
serde = { workspace = true, features = ["derive"] }
1919

2020
clap = { workspace = true, features = ["derive"] }

0 commit comments

Comments
 (0)