Skip to content

Mnist-Example#191

Open
kordjamshidi wants to merge 1 commit intodevelopfrom
mnist
Open

Mnist-Example#191
kordjamshidi wants to merge 1 commit intodevelopfrom
mnist

Conversation

@kordjamshidi
Copy link
Member

No description provided.

@kordjamshidi kordjamshidi changed the base branch from master to develop September 15, 2020 16:07
nn.Linear(hidden_sizes[1], output_size),
nn.LogSoftmax(dim=1))
def forward(self, x):
return self.recognition(x)
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why this code is needed here for this classification? Isn't there a learning module that can be used here? @guoquan @czheng17

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ModuleLearner uses a torch Module for calculating the result. Users need to define the Module anyway. You can find ModuleLearner(module=Net()) in the main code.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants