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

Move figures from .data to .utils

Browse Source
This commit is contained in:
Yin Li 2020-09-12 13:43:07 -04:00
parent 154376d95a
commit d8c6be797d
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
map2map/train.py
View File

@ -14,10 +14,9 @@ from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter from torch.utils.tensorboard import SummaryWriter
from .data import FieldDataset, DistFieldSampler from .data import FieldDataset, DistFieldSampler
from .data.figures import plt_slices, plt_power
from . import models from . import models
from .models import narrow_cast, resample, lag2eul from .models import narrow_cast, resample, lag2eul
from .utils import import_attr, load_model_state_dict from .utils import import_attr, load_model_state_dict, plt_slices, plt_power
ckpt_link = 'checkpoint.pt' ckpt_link = 'checkpoint.pt'

3
map2map/utils/__init__.py
View File

@ -1,2 +1,5 @@
from .imp import import_attr from .imp import import_attr
from .state import load_model_state_dict from .state import load_model_state_dict
from .figures import plt_slices, plt_power

0
map2map/data/figures.py → map2map/utils/figures.py
View File