thank you, .float() solved the problem! I’ll look into how to wrap the assignment using 'with torch.no_grad() ’ but could you please elaborate it if possible? My goal is to only change the weight and bias values of all layers keeping the other attributes of the layers(like drop-out functionality etc) the same as before. Will using the 'with torch.no_grad() block ’ help in attaining my goal?