train.py
from tensorboardX import SummaryWriter
from opts import parser
args=parser.parse_args()
model_name=args.model_name
a=datetime.datetime.now()
b=datetime.datetime.strftime(a,'-%Y-%m-%d-%H-%M-%S')
n_model_name=model_name+b
writer = SummaryWriter('runs/'+model_name+'/'+n_model_name)
writer.add_scalar('data/trainloss', epoch_loss, epoch)
writer.add_scalar('data/trainacc', epoch_acc, epoch)
writer.add_scalar('data/valloss', epoch_loss, epoch)
writer.add_scalar('data/valacc', epoch_acc, epoch)
writer.export_scalars_to_json("./all_scalars.json")
writer.close()
opts.py
import argparse
parser = argparse.ArgumentParser(description="PyTorch implementation of Temporal Segment Networks")
parser.add_argument('--model_name',type=str)
watch
tensorboard --logdir runs --port 66