Add _desired_num_gpus
attribute to CudaDevicePlacementMixin
#10
Job | Run time |
---|---|
49s | |
49s |
_desired_num_gpus
attribute to CudaDevicePlacementMixin
#10
Job | Run time |
---|---|
49s | |
49s |