@Poeroz did you resolve? @leoxiaobin i want to ask this line
writer_dict['writer'].add_graph(model, (dump_input, ))
is also making problem in my train.py
File "/home/anaconda3/envs/new_train/lib/python3.7/site-packages/torch/jit/init.py", line 1710, in init
assert(isinstance(orig, torch.nn.Module))
AssertionError
giving assertion error,,,,not known why?
Sorry I haven't resolved this problem yet...
Originally posted by @Poeroz in https://github.com/leoxiaobin/deep-high-resolution-net.pytorch/issues/133#issuecomment-556911840