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

Add plt.close to avoid matplotlib warning

Browse Source
This commit is contained in:
Yin Li 2020-07-28 07:10:46 -07:00
parent a886e53c54
commit 0d41bdae26
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/data/figures.py
View File

@ -12,6 +12,8 @@ from matplotlib.cm import ScalarMappable
def plt_slices(*fields, size=64, title=None, cmap=None, norm=None): def plt_slices(*fields, size=64, title=None, cmap=None, norm=None):
"""Plot slices of fields of more than 2 spatial dimensions. """Plot slices of fields of more than 2 spatial dimensions.
""" """
plt.close('all')
fields = [field.detach().cpu().numpy() if isinstance(field, torch.Tensor) fields = [field.detach().cpu().numpy() if isinstance(field, torch.Tensor)
else field for field in fields] else field for field in fields]