Fix reference to torch dataloader batch_size

NVIDIA-Merlin / Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.

https://nvidia-merlin.github.io/Transformers4Rec/main

Apache License 2.0

1.07k stars 142 forks source link

Fix reference to torch dataloader batch_size #737

Closed hese1 closed 12 months ago

hese1 commented 12 months ago

Fixes # (issue) AttributeError: 'Loader' object has no attribute '_batch_size'. Did you mean: 'batch_size'?

Goals :soccer:

Fixes a breaking bug in torch.trainer when using a dataloader

Implementation Details :construction:

Testing Details :mag:

rapids-bot[bot] commented 12 months ago

Pull requests from external contributors require approval from a NVIDIA-Merlin organization member with write permissions or greater before CI can begin.