RchalYang / Soft-Module

Code for "Multi-task Reinforcement Learning with Soft Modularization"
https://rchalyang.github.io/SoftModule
108 stars 19 forks source link

Training set #8

Open zhangt603 opened 5 months ago

zhangt603 commented 5 months ago

Hello, when I tried to run the MT10 setup as instructed in the first line on GitHub, I noticed that each epoch takes an average of 200 seconds. At this rate, it would take around 20 days to complete. Additionally, the GPU only utilizes 1235MiB of memory. Could this be a configuration issue?