Use CudaDevicePlacementMixin
in RewardModelScore
step (#845)
#251
Job | Run time |
---|---|
1m 15s | |
1m 15s |
CudaDevicePlacementMixin
in RewardModelScore
step (#845)
#251
Job | Run time |
---|---|
1m 15s | |
1m 15s |