wenhaozhao

Results 3 issues of wenhaozhao

This feature is support load vision dataset from image-foler, like torchvision.datasets.ImageFolder. In my projects, I need to load dataset from image folder for train my model, and I found candle...

Supports specifying the Huggingface mirror repository through the environment, that is, HF_ENDPOINT

Added empty_cuda_cache method to release GPU cache. ```rust tch::torch_sys::c10_cuda::empty_cuda_cache(); ``` Implemented the same functionality as torch.cuda.empty_cache() in pytorch. Example ```rust pub fn free_gpu_memory() { #[cfg(not(target_os = "macos"))] { use tch::torch_sys;...