Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

M

macaon
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Environments
- Incidents

Snippets Groups Projects

048c959c

Commit 048c959c authored 4 years ago by Franck Dary

Downloads
- Patches
- Plain Diff

Added custom hinge loss and added explorationThreshold program argument

parent 8ec956e6

No related branches found

No related tags found

No related merge requests found

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

torch_modules/include/CustomHingeLoss.hpp 13 additions, 0 deletions

torch_modules/include/CustomHingeLoss.hpp
torch_modules/src/CustomHingeLoss.cpp 17 additions, 0 deletions

torch_modules/src/CustomHingeLoss.cpp
trainer/include/Trainer.hpp 4 additions, 3 deletions

trainer/include/Trainer.hpp
trainer/src/MacaonTrain.cpp 6 additions, 3 deletions

trainer/src/MacaonTrain.cpp
trainer/src/Trainer.cpp 25 additions, 11 deletions

trainer/src/Trainer.cpp

with 65 additions and 17 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment