Skip to content

Commit

Permalink
Update nmse clustering (NVIDIA#2445)
Browse files Browse the repository at this point in the history 
* Updating refined NMSE clustering code to run on GPU.

Signed-off-by: Taejin Park <[email protected]>

* updated evaluation accuracy decimals and fixed bugs in clustering

Signed-off-by: Taejin Park <[email protected]>

* Style fix.

Signed-off-by: Taejin Park <[email protected]>

* Doc strings added and few revisions. Reflected comments by Nithin.

Signed-off-by: Taejin Park <[email protected]>

* Refelected second round of comments, simplified code and removed pyamg from req.

Signed-off-by: Taejin Park <[email protected]>

* Style fix again

Signed-off-by: Taejin Park <[email protected]>

Co-authored-by: Nithin Rao <[email protected]>
Signed-off-by: Paarth Neekhara <[email protected]>
  • Loading branch information
@tango4j @nithinraok @paarthneekhara
2 people authored and paarthneekhara committed Sep 17, 2021
1 parent ca8eb6c commit a9706a5
Show file tree
Hide file tree
Showing 3 changed files with 387 additions and 189 deletions.
Loading

0 comments on commit a9706a5

Please sign in to comment.