hiddenlayer icon indicating copy to clipboard operation
hiddenlayer copied to clipboard

Pytorch Model Branch Duplication

Open TortoiseHam opened this issue 5 years ago • 0 comments

When a pytorch model which has multiple inputs passing through the same initial layers (ex. Siamese network), hiddenlayer draws this as 2 separate network branches rather than one branch with 2 inputs. An example network which exhibits this issue can be found at: https://github.com/fastestimator/fastestimator/blob/master/apphub/one_shot_learning/siamese_network/siamese_torch.py

TortoiseHam avatar Jul 09 '20 00:07 TortoiseHam