Closed amaanda closed 4 years ago
output1 = model(img_list=resized_img_list,label_list=label_list,get_vect=True, get_prob=True,get_epe=args.evaluate) output2 = model(img_list=resized_img_list,label_list=label_list,get_vect=True, get_prob=True,get_epe=args.evaluate) print("Equals: ", torch.equal(output1['vect'], output2['vect']))
prints
False
Is there some probabilistic layer on the network?
Yes. Some PyTorch ops are non-deterministic (e.g. torch.tensor.scatteradd). If you fix all the random seeds for Python and PyTorch, you will get identical results.
prints
Is there some probabilistic layer on the network?