Hugging Face
Models
Datasets
Spaces
Posts
Docs
Enterprise
Pricing
Log In
Sign Up
Spaces:
pengc02
/
full_gaussian_avatar
like
0
Running
App
Files
Files
Community
Fetching metadata from the HF Docker repository...
ec9a6bc
full_gaussian_avatar
/
GHA
/
lib
/
dataset
/
DataLoaderX.py
pengc02
all
ec9a6bc
5 months ago
raw
Copy download link
history
blame
198 Bytes
import
torch
from
prefetch_generator
import
BackgroundGenerator
class
DataLoaderX
(torch.utils.data.DataLoader):
def
__iter__
(
self
):
return
BackgroundGenerator(
super
().__iter__())