add external files

This commit is contained in:
Shunsuke Shibayama 2024-02-11 23:04:16 +09:00
parent 7823243dbe
commit 92aa3ea078
122 changed files with 1087 additions and 0 deletions

View file

@ -0,0 +1 @@
.data = pyimport "./data"

View file

@ -0,0 +1,9 @@
{.DataLoader;} = pyimport "./dataloader"
{.Dataset;} = pyimport "./dataset"
{
.Sampler;
.SequentialSampler;
.RandomSampler;
.SubsetRandomSampler;
.WeightedRandomSampler;
} = pyimport "./sampler"

View file

@ -0,0 +1,25 @@
torch = pyimport "torch"
dataset = pyimport "./dataset"
{Sampler;} = pyimport "./sampler"
.DataLoader: ClassType
.DataLoader <: Iterable((torch.Tensor!(_, _), torch.Tensor!(_, _)))
.DataLoader.
__call__: (
dataset: dataset.Dataset,
batch_size := Nat,
shuffle := Bool,
sampler := Sampler,
batch_sampler := Sampler,
num_workers := Nat,
collate_fn := Obj,
pin_memory := Bool,
drop_last := Bool,
timeout := Float,
worker_init_fn := Obj,
multiprocessing_context := Obj,
generator := Obj,
prefetch_factor := Nat,
persistent_workers := Bool,
pin_memory_device := Str,
) -> .DataLoader

View file

@ -0,0 +1 @@
.Dataset: ClassType

View file

@ -0,0 +1,13 @@
.Sampler: ClassType
.RandomSampler: ClassType
.RandomSampler <: .Sampler
.SequentialSampler: ClassType
.SequentialSampler <: .Sampler
.SubsetRandomSampler: ClassType
.SubsetRandomSampler <: .Sampler
.WeightedRandomSampler: ClassType
.WeightedRandomSampler <: .Sampler