Thanks for your code. I am following your work, and I found a piece of confusing code that might be a mistake. It is in the file hook.py lines 103 and 104:
up_names = ['up'] * len(model.up_blocks)down_names = ['down'] * len(model.up_blocks)
Shouldn't it probably be:
down_names = ['down'] * len(model.down_blocks)
Thanks for your code. I am following your work, and I found a piece of confusing code that might be a mistake. It is in the file
hook.py
lines 103 and 104:up_names = ['up'] * len(model.up_blocks)
down_names = ['down'] * len(model.up_blocks)
Shouldn't it probably be:down_names = ['down'] * len(model.down_blocks)