Added parameter 'context' (gpu or cpu for training)

Merge request reports

Loading