search

zudi-lin / pytorch_connectomics

PyTorch Connectomics: segmentation toolbox for EM connectomics
http://connectomics.readthedocs.io/
MIT License
169 stars 77 forks source link

Fixed incorrect import in the block __init__ file and missing '**kwargs' in transformer models #132

Closed zengyuy closed 1 year ago

zengyuy commented 1 year ago

removed 'unetr_block' import statement; add the '**kwargs' in two transformer models.