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

Cleanup, Enable plt_power by default

Browse Source
This commit is contained in:
Yin Li 2021-05-30 22:35:09 -04:00
parent c7836ad7be
commit e20a3e3f62
2 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
map2map/train.py
View File

@ -5,7 +5,6 @@ import sys
from pprint import pprint from pprint import pprint
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim import torch.optim as optim
import torch.distributed as dist import torch.distributed as dist
from torch.multiprocessing import spawn from torch.multiprocessing import spawn
@ -323,10 +322,10 @@ def train(epoch, loader, model, criterion,
logger.add_figure('fig/train', fig, global_step=epoch+1) logger.add_figure('fig/train', fig, global_step=epoch+1)
fig.clf() fig.clf()
#fig = plt_power(input, lag_out, lag_tgt, label=['in', 'out', 'tgt'], fig = plt_power(input, lag_out, lag_tgt, label=['in', 'out', 'tgt'],
# **args.misc_kwargs) **args.misc_kwargs)
#logger.add_figure('fig/train/power/lag', fig, global_step=epoch+1) logger.add_figure('fig/train/power/lag', fig, global_step=epoch+1)
#fig.clf() fig.clf()
#fig = plt_power(1.0, #fig = plt_power(1.0,
# dis=[input, lag_out, lag_tgt], # dis=[input, lag_out, lag_tgt],
@ -392,10 +391,10 @@ def validate(epoch, loader, model, criterion, logger, device, args):
logger.add_figure('fig/val', fig, global_step=epoch+1) logger.add_figure('fig/val', fig, global_step=epoch+1)
fig.clf() fig.clf()
#fig = plt_power(input, lag_out, lag_tgt, label=['in', 'out', 'tgt'], fig = plt_power(input, lag_out, lag_tgt, label=['in', 'out', 'tgt'],
# **args.misc_kwargs) **args.misc_kwargs)
#logger.add_figure('fig/val/power/lag', fig, global_step=epoch+1) logger.add_figure('fig/val/power/lag', fig, global_step=epoch+1)
#fig.clf() fig.clf()
#fig = plt_power(1.0, #fig = plt_power(1.0,
# dis=[input, lag_out, lag_tgt], # dis=[input, lag_out, lag_tgt],

6
scripts/example-train.slurm
View File

@ -2,12 +2,10 @@
#SBATCH --job-name=R2D2 #SBATCH --job-name=R2D2
#SBATCH --output=%x-%j.out #SBATCH --output=%x-%j.out
#SBATCH --partition=gpu_partition #SBATCH --partition=gpu_partition
#SBATCH --gres=gpu:4
#SBATCH --exclusive
#SBATCH --nodes=2 #SBATCH --nodes=2
#SBATCH --gres=gpu:4
#SBATCH --exclusive
#SBATCH --time=1-00:00:00 #SBATCH --time=1-00:00:00