move models and data to GPUs automatically depending on the dataset size#374

Open

kayween wants to merge 8 commits intofacebookresearch:mainfrom kayween:export-D62091553

+69-42

Commits on Aug 26, 2024

make the GP Classification Model runnable on GPUs

Kaiwen Wu
authored and
facebook-github-bot
committed
move additional tensors to gpus

Kaiwen Wu
authored and
facebook-github-bot
committed
convert np.ndarray to tensor first

Kaiwen Wu
authored and
facebook-github-bot
committed
move t_term to cuda; pass test case test_1d_classification on gpu

Kaiwen Wu
authored and
facebook-github-bot
committed
test case runs on both cpu and gpu

Kaiwen Wu
authored and
facebook-github-bot
committed

Commits on Aug 27, 2024

fix pytorch warning on tensor copy

Kaiwen Wu
authored and
facebook-github-bot
committed
use torch.float64 as the dtype

Kaiwen Wu
authored and
facebook-github-bot
committed

Commits on Sep 3, 2024

move models and data to GPUs automatically depending on the dataset size (facebookresearch#374 )

Kaiwen Wu
authored and
facebook-github-bot
committed