Aochuan

icon indicating an email [email protected]

icon company The Hong Kong University of Science and Technology (Guangzhou)

Results 2 issues of Aochuan

[FEATURE] API for sampler

3 comment

Thanks for your remarkable contribution! Is there any API for `sampler` in pytorch when getting the dataloader? e.g. torch.utils.data.WeightedRandomSampler

enhancement

ammo_cuda_ext and ammo_cuda_ext_fp8 building failed

1 comment

### System Info CPU: Intel(R) Xeon(R) Platinum 8369B, GPU: a single NVIDIA A10, Driver Version: 550.54.14, CUDA Version: 12.4, NVCC Version: 12.1.105, TensorRT-LLM Version: 0.9.0.dev2024022700, nvidia-ammo Version: 0.7.4 ### Who...

bug