Andras Palffy

Shape information is not plotted

2

``` def plot_network(self, model, hltrans, input_size = [1, 1, 36]): graph = hl.build_graph(model, torch.zeros(input_size).double()) #model = torchvision.models.vgg16() #graph = hl.build_graph(model, torch.zeros([1, 3, 224, 224])) dot=graph.build_dot() dot.format="png" im=dot.render(cleanup=True) net_img=plt.imread(im) ``` I...

model.train in eval function

1

`tools/train_and_eval.py`, Line, function `eval_one_epoch` starts with setting `model.train()`, which is incorrect in my opinion. It should be `model.eval()`.

Realtime demo

8

Dear @azinke, thank you for this repository, much needed. Did you by any chance have a look at the real time options? Seems like a hassle to start even on...