Skip to content

Commit

Permalink
Update pytorch_tabular_benchmark.py
Browse files Browse the repository at this point in the history
Add benchmark results to code comments.
  • Loading branch information
wsad1 authored Sep 24, 2024
1 parent cae7ccf commit 80c1631
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions benchmark/pytorch_tabular_benchmark.py
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

0 comments on commit 80c1631

Please sign in to comment.