search

SIMEXP / fmri_predict

predicting fmri activaties from connectome
15 stars 11 forks source link

rewrite the model function to include device information #8

Open zhangyu2ustc opened 5 years ago

zhangyu2ustc commented 5 years ago

add averge_grad to class base_model

zhangyu2ustc commented 5 years ago

rewrite training function to include average_grad

zhangyu2ustc commented 5 years ago

to select the least loaded GPU for the script: check this repo: https://github.com/bamos/setGPU

add this line to your script: ''' import setGPU '''