This shows you the differences between two versions of the page.
Both sides previous revision Previous revision | Next revision Both sides next revision | ||
cs501r_f2018:lab2 [2018/09/10 18:37] wingated |
cs501r_f2018:lab2 [2018/09/10 18:37] wingated |
||
---|---|---|---|
Line 95: | Line 95: | ||
**Step 4.** Construct | **Step 4.** Construct | ||
* a model class that inherits from “nn.Module” | * a model class that inherits from “nn.Module” | ||
- | * Check out [[https://pytorch.org/docs/stable/nn.html#torch.nn.Module]] | + | ** Check out [[https://pytorch.org/docs/stable/nn.html#torch.nn.Module]] |
- | * Your model can contain any submodules you wish -- nn.Linear is a good, easy, starting point | + | ** Your model can contain any submodules you wish -- nn.Linear is a good, easy, starting point |
* a dataset class that inherits from “Dataset” and produces samples from | * a dataset class that inherits from “Dataset” and produces samples from | ||
[[https://pytorch.org/docs/stable/torchvision/datasets.html#fashion-mnist]] | [[https://pytorch.org/docs/stable/torchvision/datasets.html#fashion-mnist]] |