Rishabh Srivastava comments

Results 1 comments of


                                            Rishabh Srivastava

Uncomment the line above (im_dim = torch.FloatTensor(dim).repeat(1,2) ), it should start working