Low accuracy when loading the model and testing

[srikanthram (Srikanth)]

@ptrblck you could check out the code written here:

Blown up gradients and loss

I have my entire code written here and the output, I am surprised to see such huge gradients and losses. I am working on this code on CIFAR-10 based on local reparameterization trick by Shayer et.al, I am unable to reproduce the results in the paper. I have been working on this for over a month but no clue where I am going wrong. Could someone help me out Please. class Ternary_batch_rel(torch.nn.Module): def __init__(self,batchnorm_size): super(Ternary_batch_rel,self).__init__() self.…