• This example shows how to run training using pytorch and ToyNet with 2 classes

• Use a dataset with 2 classes

To run the sample:

• Install rocal_pybind

python3 test_training.py <image_folder> <cpu/gpu> <batch_size>

• rocal device can be cpu/gpu.