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

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

Browse Source 
This reverts commit bbf77c9f91.
This commit is contained in:
Yin Li 2020-07-16 15:22:52 -07:00
parent 5bc099251f
commit 56f8fa932b
3 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
map2map/data/__init__.py
View File

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

3
map2map/data/sampler.py
View File

@ -1,6 +1,3 @@
"""
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 from .data import FieldDataset, GroupedRandomSampler
from .data.figures import plt_slices from .data.figures import plt_slices
from . import models from . import models
from .models import narrow_cast, resample, Lag2Eul from .models import narrow_cast, resample, Lag2Eul