Skip to content

PyTorchAlpaka: Batched inference with TensorCollections, eval and frozen model, FP16 convsersion support - #50498

Closed
EmanueleCoradin wants to merge 1 commit into
cms-sw:masterfrom
EmanueleCoradin:batched-inference-PR
Closed

PyTorchAlpaka: Batched inference with TensorCollections, eval and frozen model, FP16 convsersion support#50498
EmanueleCoradin wants to merge 1 commit into
cms-sw:masterfrom
EmanueleCoradin:batched-inference-PR

Added eval mode, autofreezing, FP16 inference and mini-batches

9f505be
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs