search

tr3e / InterGen

[IJCV 2024] InterGen: Diffusion-based Multi-human Motion Generation under Complex Interactions
https://tr3e.github.io/intergen-page/
192 stars 10 forks source link

Add folder for train.py and config the model to train on market's GPU, like RTX 3090 #40

Open bring-nirachornkul opened 2 months ago

bring-nirachornkul commented 2 months ago
  1. Add folder : interhuman_processed for train.py
  2. Add text file : ignore_list.txt, train.txt.
  3. Config the model to train on market's GPU, like RTX 3090
bring-nirachornkul commented 2 months ago

Updated train.py to save only the latest checkpoint after each epoch using save_top_k=1, ensuring the previous checkpoint is deleted, and added a function to automatically resume from the latest checkpoint if available.