Skip to content
This repository was archived by the owner on Feb 13, 2025. It is now read-only.

Commit 8141444

Browse files
authored
Remove outdated references to Batcher from projects which no longer use that API. (#591)
1 parent 1679ff7 commit 8141444

File tree

12 files changed

+2
-15
lines changed

12 files changed

+2
-15
lines changed

Autoencoder/Autoencoder1D/main.swift

-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ import Datasets
1616
import Foundation
1717
import ModelSupport
1818
import TensorFlow
19-
import Batcher
2019

2120
let epochCount = 10
2221
let batchSize = 100

Autoencoder/Autoencoder2D/main.swift

-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ import Datasets
1818
import Foundation
1919
import ModelSupport
2020
import TensorFlow
21-
import Batcher
2221

2322
let epochCount = 10
2423
let batchSize = 100

Benchmarks/Models/ImageClassificationInference.swift

-2
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
import Batcher
1615
import Datasets
1716
import ImageClassificationModels
1817
import TensorFlow
@@ -23,7 +22,6 @@ protocol ImageClassificationModel: Layer where Input == Tensor<Float>, Output ==
2322
static var outputLabels: Int { get }
2423
}
2524

26-
// TODO: Ease the tight restriction on Batcher data sources to allow for lazy datasets.
2725
class ImageClassificationInference<Model, ClassificationDataset>: Benchmark
2826
where
2927
Model: ImageClassificationModel,

Benchmarks/Models/ImageClassificationTraining.swift

-2
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,9 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
import Batcher
1615
import Datasets
1716
import TensorFlow
1817

19-
// TODO: Ease the tight restriction on Batcher data sources to allow for lazy datasets.
2018
struct ImageClassificationTraining<Model, ClassificationDataset>: Benchmark
2119
where
2220
Model: ImageClassificationModel, Model.TangentVector.VectorSpaceScalar == Float,

Benchmarks/Models/SyntheticImageDataset.swift

-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
import Batcher
1615
import Datasets
1716
import TensorFlow
1817

Datasets/CIFAR10/CIFAR10.swift

-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
import Foundation
2121
import ModelSupport
2222
import TensorFlow
23-
import Batcher
2423

2524
public struct CIFAR10<Entropy: RandomNumberGenerator> {
2625
/// Type of the collection of non-collated batches.

Datasets/MNIST/FashionMNIST.swift

-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919

2020
import Foundation
2121
import TensorFlow
22-
import Batcher
2322

2423
public struct FashionMNIST<Entropy: RandomNumberGenerator> {
2524
/// Type of the collection of non-collated batches.

Datasets/MNIST/KuzushijiMNIST.swift

-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818

1919
import Foundation
2020
import TensorFlow
21-
import Batcher
2221

2322
public struct KuzushijiMNIST<Entropy: RandomNumberGenerator> {
2423
/// Type of the collection of non-collated batches.

Datasets/MNIST/MNIST.swift

-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919

2020
import Foundation
2121
import TensorFlow
22-
import Batcher
2322

2423
public struct MNIST<Entropy: RandomNumberGenerator> {
2524
/// Type of the collection of non-collated batches.

Datasets/TextUnsupervised/TextUnsupervised.swift

-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
import Batcher
1615
import Foundation
1716
import ModelSupport
1817
import TensorFlow

Examples/GPT2-WikiText2/main.swift

-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
import Batcher
1615
import Datasets
1716
import TensorFlow
1817
import TextModels

Package.swift

+2-2
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ let package = Package(
8686
exclude: ["UI/Windows/main.swift", "UI/macOS/main.swift"]),
8787
.target(
8888
name: "GPT2-WikiText2",
89-
dependencies: ["Batcher", "Datasets", "TextModels"],
89+
dependencies: ["Datasets", "TextModels"],
9090
path: "Examples/GPT2-WikiText2",
9191
exclude: ["UI/Windows/main.swift"]),
9292
.testTarget(name: "TextTests", dependencies: ["TextModels"]),
@@ -112,7 +112,7 @@ let package = Package(
112112
.testTarget(name: "SupportTests", dependencies: ["ModelSupport"]),
113113
.target(
114114
name: "CycleGAN",
115-
dependencies: ["Batcher", "ArgumentParser", "ModelSupport", "Datasets"],
115+
dependencies: ["ArgumentParser", "ModelSupport", "Datasets"],
116116
path: "CycleGAN"
117117
),
118118
.target(

0 commit comments

Comments
 (0)