Popular repositories Loading

1. gpu_memory_profiling gpu_memory_profiling Public

   Profile the GPU memory usage of every line in a Pytorch code

   Python 82 19

2. Train_with_mixed_batch_keras Train_with_mixed_batch_keras Public

   Train a multi-task learning model to estimate age, gender and emotion of an input face

   Python 6 6

3. setGPU setGPU Public

   Forked from bamos/setGPU

   Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.

   Python 5 2

4. handy_scripts handy_scripts Public

   Python 4 2

5. line_profiler_example line_profiler_example Public

   Python 4

6. Face_age_gender_emotion Face_age_gender_emotion Public

   Training on multiple datasets with ''mixed batches'' strategy

   Python 3 2