n0obcoder · August 28, 2019 20:28
diff --git a/fitting_straight_line_pytorch___07___loss_fn_and_optimizer.py b/fitting_straight_line_pytorch___07___loss_fn_and_optimizer.py
 # Defining the Loss Function
 # Mean Squared Error is the most common choice of Loss Function for Linear Regression models.
 criterion = torch.nn.MSELoss()

 # Defining the Optimizer, which would update all the trainable parameters of the model, making the model learn the data distribution better and hence fit the distribution better.
 optimizer = torch.optim.Adam(model.parameters(), lr = 0.0005)
	# Defining the Loss Function
	# Mean Squared Error is the most common choice of Loss Function for Linear Regression models.
	criterion = torch.nn.MSELoss()

	# Defining the Optimizer, which would update all the trainable parameters of the model, making the model learn the data distribution better and hence fit the distribution better.
	optimizer = torch.optim.Adam(model.parameters(), lr = 0.0005)
No results found