Skip to content

Update - #8

Merged
swethmandava merged 36 commits into
swethmandava:masterfrom
NVIDIA:master
May 13, 2020
Merged

Update#8
swethmandava merged 36 commits into
swethmandava:masterfrom
NVIDIA:master

Conversation

@swethmandava

Copy link
Copy Markdown
Owner

No description provided.

pribalta and others added 30 commits April 8, 2020 09:40
Signed-off-by: Pablo Ribalta <pribalta@nvidia.com>
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
- Revert network to use fcPlugin and geluPlugin

Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
Replace softmax_cross_entropy_with_logits with binary_crossentropy
constant batch in TensorRT engine build
Intel's optimization for PyTorch on CPU are added, you need to set "export OMP_NUM_THREADS=num physical cores" based on your CPU's core number
add Intel optimization for PyTorch
@swethmandava
swethmandava merged commit edef331 into swethmandava:master May 13, 2020
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.

10 participants