Héctor Martel
Héctor Martel
Hi @zqlsnr , I will try to reproduce the issue and get back to you
I was experiencing the same problem and this solution worked for me: ```python from iunets import iUNet model = iUNet( in_channels=64, architecture=[2,2,2,2], dim=2 ) model.print_layout() ``` This example is in...
Pretrained models will be provided shortly. Please see the answers to your other issues: * #5 * #4
This checkpoint is obtained by training the frame auto-encoder that is defined [here](https://github.com/hmartelb/avlit/blob/main/src/modules/autoencoder.py). The pretrained models will be available soon. Please refer to our paper regarding how to train this...