Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update pytorch_tabular_benchmark.py #457

Merged
merged 7 commits into from
Oct 2, 2024
Merged

Update pytorch_tabular_benchmark.py #457

Changes from 2 commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
80c1631
Update pytorch_tabular_benchmark.py
wsad1 Sep 24, 2024
bbb60ca
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Sep 24, 2024
d48382d
Update README.md
wsad1 Sep 30, 2024
700da76
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Sep 30, 2024
dd63def
Merge branch 'master' into wsad1-patch-1
wsad1 Sep 30, 2024
463631e
Update README.md
yiweny Oct 2, 2024
dcd7220
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 2, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 12 additions & 0 deletions benchmark/pytorch_tabular_benchmark.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,17 @@
"""This script benchmarks the training time of TabTransformer using PyTorch
Frame and PyTorch Tabular.

Results form comparing Pytorch Tabular and Frame. Specifically the iteration
speed while trainig.

-------------------------------------
Package | Model | Num iters/sec|
-------------------------------------
Tabular | TabNet | 41.7
Frame | TabNet | 45.0
Tabular | FTTrans | 40.1
Frame | FTTrans | 43.7
--------------------------------------
"""
import argparse
import os.path as osp
Expand Down
Loading