n0obcoder · November 21, 2019 18:10
diff --git a/mobile_gallery_pytorch_07_loss_func_and_optimizer.py b/mobile_gallery_pytorch_07_loss_func_and_optimizer.py
 import torch.optim as optim
 from torch.optim import lr_scheduler

 # Now we define the Loss Function
 loss_fn = nn.CrossEntropyLoss()

 # Define the optimizer
 lr = 0.00001
 optimizer = optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr = lr)
	import torch.optim as optim
	from torch.optim import lr_scheduler

	# Now we define the Loss Function
	loss_fn = nn.CrossEntropyLoss()

	# Define the optimizer
	lr = 0.00001
	optimizer = optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr = lr)
No results found