Can't visualize nn.Contiguous

[ahmedmagdiosman]

Hello, awesome work with this package! I have an issue trying to visualize nn.Contiguous. Do you know what might be causing it?
If it's not an easy fix, is there a way to ignore this node in visualization?

Edit: This also occurs with nn.SoftMax. I would think it would work with SoftMax. Am I doing something wrong?
Edit2: I think this happens whenever the previous module is a SplitTable or ParallelTable.

stack traceback:
	[C]: in function 'assert'
	...e/aosman/torch/install/share/lua/5.1/optnet/graphgen.lua:171: in function 'addEdge'
	...e/aosman/torch/install/share/lua/5.1/optnet/graphgen.lua:196: in function <...e/aosman/torch/install/share/lua/5.1/optnet/graphgen.lua:190>
	[C]: in function 'xpcall'
	/home/aosman/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
	.../aosman/torch/install/share/lua/5.1/nn/ParallelTable.lua:12: in function <.../aosman/torch/install/share/lua/5.1/nn/ParallelTable.lua:10>
	[C]: in function 'xpcall'
	/home/aosman/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
	/home/aosman/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function </home/aosman/torch/install/share/lua/5.1/nn/Sequential.lua:41>
	[C]: in function 'xpcall'
	...
	[C]: in function 'xpcall'
	/home/aosman/torch/install/share/lua/5.1/itorch/main.lua:210: in function </home/aosman/torch/install/share/lua/5.1/itorch/main.lua:174>
	/home/aosman/torch/install/share/lua/5.1/lzmq/poller.lua:75: in function 'poll'
	/home/aosman/torch/install/share/lua/5.1/lzmq/impl/loop.lua:307: in function 'poll'
	/home/aosman/torch/install/share/lua/5.1/lzmq/impl/loop.lua:325: in function 'sleep_ex'
	/home/aosman/torch/install/share/lua/5.1/lzmq/impl/loop.lua:370: in function 'start'
	/home/aosman/torch/install/share/lua/5.1/itorch/main.lua:389: in main chunk
	[C]: in function 'require'
	(command line):1: in main chunk
	[C]: at 0x00405d50```