Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Deprecate GroupedRandomSampler for use of memmap in place of cache

Browse Source
This commit is contained in:
Yin Li 2020-07-14 18:47:26 -04:00
parent 5fd8cbfcda
commit bbf77c9f91
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
map2map/data/__init__.py
View File

@ -1,2 +1 @@
from .fields import FieldDataset from .fields import FieldDataset
from .sampler import GroupedRandomSampler

3
map2map/data/sampler.py
View File

@ -1,3 +1,6 @@
"""
Deprecated for use of memmap in place of cache
"""
from itertools import chain from itertools import chain
import torch import torch
from torch.utils.data import Sampler from torch.utils.data import Sampler

2
map2map/train.py
View File

@ -14,7 +14,7 @@ from torch.utils.data.distributed import DistributedSampler
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter from torch.utils.tensorboard import SummaryWriter
from .data import FieldDataset, GroupedRandomSampler from .data import FieldDataset
from .data.figures import plt_slices from .data.figures import plt_slices
from . import models from . import models
from .models import (narrow_cast, resample, from .models import (narrow_cast, resample,