Closed punithsekar closed 3 weeks ago
Reference link:- https://github.com/mateuszbuda/brain-segmentation-pytorch/tree/d45f8908ab2f0246ba204c702a6161c9eb25f902
Vanilla Unet torch implementation of our reference implementation - https://github.com/tenstorrent/tt-metal/blob/05a8838c0d935d41112e07bc867b4cfc40115d09/models/experimental/functional_vanilla_unet/reference/unet.py
Our pytorch model graph - model_vanilla_unet.gv.pdf
Modules:
Ops:
fyi @saichandax
Closing this for now, We can open it if needed!
Reference link:- https://github.com/mateuszbuda/brain-segmentation-pytorch/tree/d45f8908ab2f0246ba204c702a6161c9eb25f902
Vanilla Unet torch implementation of our reference implementation - https://github.com/tenstorrent/tt-metal/blob/05a8838c0d935d41112e07bc867b4cfc40115d09/models/experimental/functional_vanilla_unet/reference/unet.py
Our pytorch model graph - model_vanilla_unet.gv.pdf
Modules:
Ops: