Revert "Deprecate GroupedRandomSampler for use of memmap in place of cache"

This reverts commit bbf77c9f91.
This commit is contained in:
Yin Li 2020-07-18 18:07:38 -04:00
parent 2d5234812b
commit 509856cb4f
3 changed files with 2 additions and 4 deletions

View File

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

View File

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

View File

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