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

Merge branch 'master' into lag2eul

Browse Source
This commit is contained in:
Yin Li 2020-07-14 18:47:54 -04:00
commit 6fa85f8285
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_like, resample, Lag2Eul, from .models import (narrow_like, resample, Lag2Eul,