Skip to content

GitLab

Unverified Commit db047d13 authored by Benjamin Lefaudeux's avatar Benjamin Lefaudeux Committed by GitHub
Browse files

[bug] OSS README typo #2

parent 656828bb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ db047d13
...@@ -39,7 +39,7 @@ def train( ...@@ -39,7 +39,7 @@ def train(
# Problem statement # Problem statement
model = myAwesomeModel() model = myAwesomeModel()
dataloader = mySuperFastDataloader() dataloader = mySuperFastDataloader()
loss = myVeryRelevantLoss() loss_fn = myVeryRelevantLoss()
base_optimizer = torch.optim.SGD # pick any pytorch compliant optimizer here base_optimizer = torch.optim.SGD # pick any pytorch compliant optimizer here
base_optimizer_arguments = {} # pass any optimizer specific arguments here, or directly below when instantiating OSS base_optimizer_arguments = {} # pass any optimizer specific arguments here, or directly below when instantiating OSS
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment