fix multi-gpu - Githubissues

cognitivecomputations / laserRMT

This is our own implementation of 'Layer Selective Rank Reduction'

Apache License 2.0

229 stars 26 forks source link

Open ehartford opened 8 months ago

ehartford commented 8 months ago

Currently, the job is not split across all gpus. (in rmt_laser_snr_math.py) Need to scale this to run across all GPUs

fernando-neto-ai commented 4 months ago

Still working on that