IMTA icon indicating copy to clipboard operation
IMTA copied to clipboard

Results 2 IMTA issues
Sort by recently updated
recently updated
newest added

Hi, kalviny. Thank you very much for this repo! Would you share the pretrained 4-scale MSDNet w.o./w GE/IMTA on Imagenet? Training on imagenet takes a long time. In the paper,...

At the forward method on the GradientRescaleFunction, the input tensor is saved to be used in the backward function: https://github.com/kalviny/IMTA/blob/a9378373241c3247db01f826e3cd8025702fa54b/models/msdnet_ge.py#LL11C6-L11C6 and at the backward function, the saved input is retrieved:...