Added tensorboard instructions

README.md

@@ -158,6 +158,12 @@ There is also [Visdom](https://github.com/facebookresearch/visdom) support to vi
 python train.py --visdom
 ```
 
+There is also [Tensorboard](https://github.com/lanpa/tensorboard-pytorch) support to visualise training. Follow the instructions to set up. To use:
+
+```
+python train.py --tensorboard --logdir log_dir/ # Make sure the tensorboard instance is made pointing to this log directory
+```
+
 ### Noise Augmentation/Injection
 
 There is support for two different types of noise; noise augmentation and noise injection.

train.py

@@ -99,7 +99,7 @@ def main():
  os.makedirs(args.log_dir)
  except OSError as e:
  if e.errno == errno.EEXIST:
- print('Directory already exists.')
+ print('Tensorboard log directory already exists.')
  for file in os.listdir(args.log_dir):
  file_path = os.path.join(args.log_dir, file)
  try:
@@ -116,7 +116,7 @@ def main():
  os.makedirs(save_folder)
  except OSError as e:
  if e.errno == errno.EEXIST:
- print('Directory already exists.')
+ print('Model Save directory already exists.')
  else:
  raise
  criterion = CTCLoss()