Open sinagho opened 6 days ago
It is integrated into the LHUNet model (v12). To do this, I added moga.py in the cnn folder (also __init__.py) and added a new code M for vit blocks to link it with the moga3d (here).
v12
moga.py
cnn
__init__.py
M
You can check the model using check_lhunet.ipynb
check_lhunet.ipynb