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

Sequence primitive for neural network layers.

    XMLWordPrintableJSON

Details

    • New Feature
    • Status: Open
    • Major
    • Resolution: Unresolved
    • None
    • None
    • APIs, PythonAPI

    Description

      This JIRA task is to construct a sequence primitive for the python API that is able to combine multiple neural network layers, and perform forward and backwards passes.

      The result should be similar to Keras and PyTorch (not forcefully similar, but a nice design like theirs.)

      from tensorflow.keras import layers, models
      p=’same’
      model = models.Sequential([
      layers.Conv2D(32, (3, 3), activation=’relu’, padding=p, input_shape=(H, W, C)),
      layers.Conv2D(64, (3, 3), activation=’relu’, padding=p),
      layers.Conv2D(128, (3, 3), activation=’relu’, padding=p),
      layers.Conv2D(1, (1, 1), padding=p)])
      Or pytorch:
      import torch.nn as nn
      model = nn.Sequential(
      nn.Conv2d(C, 32, kernel_size=3, padding=1), nn.ReLU(),
      nn.Conv2d(32, 64, kernel_size=3, padding=1), nn.ReLU(),
      nn.Conv2d(64, 128, kernel_size=3, padding=1), nn.ReLU(),
      nn.Conv2d(128, 1, kernel_size=1, padding=0))

      Attachments

        Activity

          People

            Unassigned Unassigned
            baunsgaard Sebastian Baunsgaard
            Votes:
            0 Vote for this issue
            Watchers:
            1 Start watching this issue

            Dates

              Created:
              Updated: