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.

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

PyTorch Data-loader with Rust-backend. #117

Merged
merged 123 commits into from
Mar 20, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
123 commits
Select commit Hold shift + click to select a range
6ab014a
Add Pytorch dataloader.
benjaminrwilson Oct 18, 2022
7eaf448
Updates.
benjaminrwilson Oct 18, 2022
303f405
Update tests.
benjaminrwilson Oct 18, 2022
9b4be05
Update ci.yml
benjaminrwilson Oct 20, 2022
183e871
Update environment.yml
benjaminrwilson Oct 20, 2022
d7e4b36
Update setup.cfg
benjaminrwilson Oct 20, 2022
b92eb1b
Update noxfile.py
benjaminrwilson Oct 20, 2022
c4aee54
Fix nits.
Oct 20, 2022
c912838
Fix nits + polars leftovers.
Oct 20, 2022
8d74b27
Remove caching.
benjaminrwilson Nov 7, 2022
355d677
Remove sorting.
benjaminrwilson Nov 7, 2022
98f55a1
Fix non-polars calls.
benjaminrwilson Nov 23, 2022
b771c16
Update default columns.
benjaminrwilson Nov 23, 2022
3fce01d
Merge branch 'main' into feature/torch-dataloader-pandas
benjaminrwilson Feb 8, 2023
e639dc9
Merge branch 'main' into feature/torch-dataloader-pandas
benjaminrwilson Feb 27, 2023
af894d4
Merge branch 'main' into feature/torch-dataloader-pandas
benjaminrwilson Feb 27, 2023
b291c13
Rust integration prototype.
benjaminrwilson Mar 11, 2023
cd16b3b
Add rust/torch integration.
benjaminrwilson Mar 11, 2023
c2f283a
Add docstrings.
benjaminrwilson Mar 11, 2023
4a7a18d
Integrate sweep type return.
benjaminrwilson Mar 11, 2023
ed8f27b
Update deps.
benjaminrwilson Mar 11, 2023
8d99fbf
Fix clippy.
benjaminrwilson Mar 11, 2023
6e2941e
Add workspace.
benjaminrwilson Mar 11, 2023
b835f36
Add voxelization op.
benjaminrwilson Mar 11, 2023
af49d42
Improve example code.
benjaminrwilson Mar 12, 2023
c08134b
Add track_uuid to rust dataloader.
benjaminrwilson Mar 13, 2023
11a4791
Return city pose from torch dataloader.
benjaminrwilson Mar 13, 2023
baf9ae0
Update deps.
benjaminrwilson Mar 13, 2023
8147529
Remove unused dependencies.
benjaminrwilson Mar 13, 2023
afe9baf
Add scene flow dataloader.
benjaminrwilson Mar 13, 2023
a7103f3
Remove imports / comments.
benjaminrwilson Mar 13, 2023
b8407ba
Use next index.
benjaminrwilson Mar 13, 2023
67e05c9
Remove init check.
benjaminrwilson Mar 13, 2023
667086c
Remove unused args.
benjaminrwilson Mar 13, 2023
9ffbf87
Use from rust in sensor dataloader.
benjaminrwilson Mar 13, 2023
59c4f2b
Fix typing issues.
benjaminrwilson Mar 13, 2023
e335594
Change example default to 1 accum sweep.
benjaminrwilson Mar 13, 2023
ab9abbf
Switch from lazy to eager loading.
benjaminrwilson Mar 13, 2023
cdf9253
Add memory_mapped flag.
benjaminrwilson Mar 13, 2023
8589912
Add docstrings.
benjaminrwilson Mar 13, 2023
aadb725
Add docstrings.
benjaminrwilson Mar 13, 2023
1de301d
Add type.
benjaminrwilson Mar 13, 2023
2d12c30
Remove unused import.
benjaminrwilson Mar 13, 2023
919fd8a
Fix lint.
benjaminrwilson Mar 13, 2023
0c62256
Update naming + docstrings.
benjaminrwilson Mar 13, 2023
415dda1
r -> rust.
benjaminrwilson Mar 13, 2023
f7c2f68
Minor docstring fixes.
benjaminrwilson Mar 13, 2023
6723578
Fix lint.
benjaminrwilson Mar 14, 2023
4631535
Fix lint.
benjaminrwilson Mar 14, 2023
3be0f60
Remove test data.
benjaminrwilson Mar 14, 2023
0be4d6d
Remove pandas stubs.
benjaminrwilson Mar 14, 2023
fcbd2f1
Add docstrings.
benjaminrwilson Mar 14, 2023
07140a0
Add docstrings.
benjaminrwilson Mar 14, 2023
de6ba7a
Reorganize files.
benjaminrwilson Mar 14, 2023
8e50166
Cleanup dataloader get method.
benjaminrwilson Mar 14, 2023
9e30925
Minor fixes on docstrings.
benjaminrwilson Mar 14, 2023
9df40e0
Fix args ordering.
benjaminrwilson Mar 14, 2023
688c4b1
Add additional comments.
benjaminrwilson Mar 14, 2023
3f6848c
Update docstring.
benjaminrwilson Mar 14, 2023
02d6804
Add image deps.
benjaminrwilson Mar 14, 2023
e2848c3
Update arg.
benjaminrwilson Mar 14, 2023
1e02e33
Update readmes.
benjaminrwilson Mar 14, 2023
c6146d1
Add additional readme + remove unused code.
benjaminrwilson Mar 14, 2023
a51e369
Fix lint.
benjaminrwilson Mar 14, 2023
e6d1a97
Add option type for annotations.
benjaminrwilson Mar 14, 2023
afe0cf7
Remove hard-coded mmap.
benjaminrwilson Mar 14, 2023
19384c9
Update readmes.
benjaminrwilson Mar 14, 2023
5c560bd
Update torch init.
benjaminrwilson Mar 14, 2023
07b7eff
Fix utils shapes.
benjaminrwilson Mar 15, 2023
e5fea8a
Update typing.
benjaminrwilson Mar 15, 2023
2309934
Modify docstring.
benjaminrwilson Mar 15, 2023
9baffc6
Update utils.py
benjaminrwilson Mar 15, 2023
c618359
Add maximize build space.
benjaminrwilson Mar 17, 2023
fdf6ae0
Merge branch 'feature/rust' of https://github.com/argoverse/av2-api i…
benjaminrwilson Mar 17, 2023
027b21a
Comment out torch temporarily.
benjaminrwilson Mar 17, 2023
455baca
Restructure torch dataloader.
benjaminrwilson Mar 17, 2023
6344744
Update ci.
benjaminrwilson Mar 17, 2023
c37f2f4
Update ci.
benjaminrwilson Mar 17, 2023
f62cb09
Remove explicit final types.
benjaminrwilson Mar 18, 2023
f82c083
Update object types.
benjaminrwilson Mar 19, 2023
e6d1d78
Docstring improvements.
benjaminrwilson Mar 19, 2023
bf1ef2a
Update dataloader helpers.
benjaminrwilson Mar 19, 2023
708caf6
Update dataloader helpers.
benjaminrwilson Mar 19, 2023
b5ab938
Minor updates on cuboids.
benjaminrwilson Mar 19, 2023
65483d5
Update detection dataloader + rust nits.
benjaminrwilson Mar 19, 2023
cb8bcd8
Fix lint.
benjaminrwilson Mar 19, 2023
4b836c9
Fix typing.
benjaminrwilson Mar 19, 2023
3d0ff7c
Fix docstrings.
benjaminrwilson Mar 19, 2023
2c5bbe1
Update column naming.
benjaminrwilson Mar 19, 2023
3f64f01
Add additional examples.
benjaminrwilson Mar 19, 2023
8740374
Add additional examples.
benjaminrwilson Mar 19, 2023
ee51df8
Update torch structures.
benjaminrwilson Mar 19, 2023
d2feca1
Update se3.rs
benjaminrwilson Mar 19, 2023
6882596
Add unit tests.
benjaminrwilson Mar 19, 2023
98315a0
Merge branch 'feature/rust' of https://github.com/argoverse/av2-api i…
benjaminrwilson Mar 19, 2023
a6924ed
Add additional tests.
benjaminrwilson Mar 19, 2023
4728336
Update naming.
benjaminrwilson Mar 19, 2023
9df5c5c
Update rust vars.
benjaminrwilson Mar 19, 2023
e3bb48a
Update naming.
benjaminrwilson Mar 19, 2023
2a91e43
Fix nit
benjaminrwilson Mar 19, 2023
f9b020f
Fix docstring.
benjaminrwilson Mar 19, 2023
55915ae
Fix docstring.
benjaminrwilson Mar 19, 2023
5a8d024
Fix enum.
benjaminrwilson Mar 19, 2023
502842f
Add utils tests.
benjaminrwilson Mar 19, 2023
3fd3a31
Update README.md
benjaminrwilson Mar 19, 2023
28fa6f1
Add additional tests.
benjaminrwilson Mar 19, 2023
aaffaae
Remove dead code.
benjaminrwilson Mar 19, 2023
bec2fee
Merge branch 'feature/rust' of https://github.com/argoverse/av2-api i…
benjaminrwilson Mar 19, 2023
b1eb02b
Update spelling.
benjaminrwilson Mar 19, 2023
7d08883
Update docstrings.
benjaminrwilson Mar 19, 2023
96aae61
Add additional examples.
benjaminrwilson Mar 19, 2023
663e80c
Remove unused import.
benjaminrwilson Mar 19, 2023
b2d9a47
Update versioning.
benjaminrwilson Mar 19, 2023
a5b9bac
Rename rust methods.
benjaminrwilson Mar 19, 2023
50d395f
Update docstrings.
benjaminrwilson Mar 19, 2023
82687cf
Update sweep variables.
benjaminrwilson Mar 19, 2023
1b76576
Minor docstring updates.
benjaminrwilson Mar 19, 2023
02ee74b
Add rust warning.
benjaminrwilson Mar 19, 2023
19b5854
Minor docstring updates.
benjaminrwilson Mar 19, 2023
ca24f4f
Add init files to test.
benjaminrwilson Mar 20, 2023
65c9176
Update scene flow.
benjaminrwilson Mar 20, 2023
c3f66f2
Add docstrings + change assert_allclose.
benjaminrwilson Mar 20, 2023
093faa3
Testing fixes.
benjaminrwilson Mar 20, 2023
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
7 changes: 7 additions & 0 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,13 @@ jobs:
miniforge-version: latest
python-version: ${{ matrix.python_version }}

- name: Reclaim space.
run: |
rm -rf /usr/share/dotnet
rm -rf /opt/ghc
rm -rf "/usr/local/share/boost"
rm -rf "$AGENT_TOOLSDIRECTORY"

- name: Install prerequisites.
run: |
mamba install -y nox pip pyyaml
Expand Down
Loading