havakv / havakv/pycox

Is there any way around model.fit(x_train, y_train)?

Open
#148 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
995
Forks
203
PR merge metrics
No merged PRs in 30d

Description

Because x_train is too large to be uploaded on GPU (RTX 3090, 24GB), calling model.fit (x_train, y_train) cannot train the model.

Is there any way around instead of using model.fit (x_train, y_train) ? (e.g. using conventional Pytorch training such as DataLoader with a small batch ?)

Thank you in advance:)

Contributor guide

No contributing guide indexed for this repository

Research direction

The issue names model.fit(x_train, y_train) and a conventional PyTorch DataLoader with small batches, but no files or tests. Start by locating the model.fit training entry point and its batching documentation; done means establishing and documenting a supported way to train when the full dataset does not fit on the GPU.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.