Skip to content

Commit 04efeaf

Browse files
committed
revert some
1 parent 0009e27 commit 04efeaf

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/TorchSharp/DataLoader.cs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
using System;
44
using System.Collections;
55
using System.Collections.Generic;
6-
using System.Diagnostics.CodeAnalysis;
76
using System.Diagnostics.SymbolStore;
87
using System.Linq;
98
using System.Threading;
@@ -182,7 +181,7 @@ public class DataLoader<T, S> : IEnumerable<S>, IDisposable
182181
/// <param name="batchSize">Size of batch</param>
183182
/// <param name="collate_fn">Callback to merge items make to a batch</param>
184183
/// <param name="device">device for output tensor</param>
185-
/// <param name="shuffler">Shuffler for dataloader.</param>
184+
/// <param name="shuffler">Shuffler for dataloader</param>
186185
/// <param name="num_worker">Count of worker</param>
187186
/// <param name="drop_last">
188187
/// Set to true to drop the last incomplete batch, if the dataset size is not divisible by the batch size.

src/TorchSharp/Utils/TensorDataset.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ public override long Count {
6262
return this[index];
6363
}
6464

65-
private torch.Tensor[] _tensors;
65+
readonly torch.Tensor[] _tensors;
6666

6767
protected override void Dispose(bool disposing)
6868
{

0 commit comments

Comments
 (0)