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

Cleanup preliminary ML code #19

Merged
merged 29 commits into from
Oct 31, 2024
Merged

Cleanup preliminary ML code #19

merged 29 commits into from
Oct 31, 2024

Conversation

Flova
Copy link
Member

@Flova Flova commented Oct 26, 2024 •
edited
Loading

Summary

Don't be scared of the large number of changes... The preliminary testing scripts have a lot of duplication, and they are moved in this PR. Also, the additional dependencies add a many lines to the lock file (1.7k additions).

  • Moves the current state of the preliminary testing scripts to their own folder for later reference as they are split up into the final architecture
  • Split the most promising model (context diffusion) from the preliminary tests into different components
  • Begin creation of a training script
  • Overall formatting of the ML part
  • Usage of poetry for the ML deps

Related issues

Flova added 28 commits October 24, 2024 21:52
@Flova
Remove outdated files
56dd346
@Flova
Move files
216d264
@Flova
Downgrade python version for 22.04 support
d7f1c80
@Flova
Add ml deps
b583bd0
@Flova
Apply formatting
5283388
@Flova
Working version of chunced prediction
0f13210
@Flova
Add ema and convergence plots
1a52495
@Flova
Add runner
8cdad13
@Flova
Improve diffusion transformer
c40827d
@Flova
Remove causality from model
fd012ca
@Flova
Cleanup
3f3dd9d
@Flova
Apply formatting
df2b47e
@Flova
Rename outdated variable
d9ede24
@Flova
Train diffusion model on robot data
c75871a
@Flova
Add weights to gitignore
b4cc464
@Flova
Apply formatting
b4a146a
@Flova
Run with all joints
43b1005
@Flova
Normalize radians
1c2ad76
@Flova
Increase epochs and batch size
007a577
@Flova
Cleanup old notation
caeedbe
@Flova
Add context diffusion transformer
40e1b98
@Flova
Add norm info
c0d836d
@Flova
Dont clip
0f6fd8b
@Flova
Add context transformer
b6ab2ce
@Flova
Add better norm
8677c04
@Flova
add denoising plots
93177a0
@Flova
Move preliminary ml stuff to it's own folder
b6d59e8
@Flova
Update structure
c7e6dad
@Flova Flova self-assigned this Oct 29, 2024
@Flova Flova marked this pull request as ready for review October 29, 2024 14:16
@Flova Flova changed the title Cleanup ML stuff Cleanup preliminary ML code Oct 29, 2024
@Flova Flova requested review from jaagut and texhnolyze October 29, 2024 14:17
@jaagut jaagut merged commit 9298b46 into main Oct 31, 2024
2 checks passed
@jaagut jaagut deleted the cleanup branch October 31, 2024 10:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@jaagut jaagut jaagut approved these changes

@texhnolyze texhnolyze Awaiting requested review from texhnolyze

Assignees

@Flova Flova

Labels
None yet
Projects
DDLitLab
Status: Done
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

Vorlaeufige tests verschiedener Modell typen (RNN, Diffusion, Discrete)
2 participants
@Flova @jaagut