Support float trainable masks #10
Description
Great work! How can I change the code so that is supports masks that are floats, not only integer/boolean? I need also to calculate the gradient of the mask, how do I do this?
{{ message }}
Great work! How can I change the code so that is supports masks that are floats, not only integer/boolean? I need also to calculate the gradient of the mask, how do I do this?