Uploaded image for project: 'SystemDS'
  1. SystemDS
  2. SYSTEMDS-1681

Add a unit test for 2D convolution comparing against another system

    XMLWordPrintableJSON

Details

    • New Feature
    • Status: In Progress
    • Major
    • Resolution: Unresolved
    • None
    • None
    • None
    • None

    Description

      We should add a comparison against TensorFlow in nn/test/test.dml, like we have for other layers, and check that the same values are produced on CPU & GPU for both TensorFlow and SystemML. There is already a conv2d function in nn/test/test.dml that compares the output of three different 2D convolution layers that we have (built-in, DML im2col, DML basic). We should add a new section to this conv2d function (similar to how the max_pool2d test function has multiple test sections) to compare the built-in conv2d layer against the output from TensorFlow. We should use the same X, W, and b initialization values as used in conv2d_transpose.

      Attachments

        Activity

          People

            dusenberrymw Mike Dusenberry
            dusenberrymw Mike Dusenberry
            Votes:
            0 Vote for this issue
            Watchers:
            1 Start watching this issue

            Dates

              Created:
              Updated: