Search Unity

Bug When barracuda support torch.nn.Flatten layer?

Discussion in 'Barracuda' started by lijianfeng, Jan 27, 2021.

  1. lijianfeng

    lijianfeng

    Joined:
    Sep 8, 2015
    Posts:
    54
    model=torch.nn.Sequential(
    torch.nn.Linear(2,1),
    torch.nn.Flatten(0)
    )

    this model can't be imported to unity,it looks like current doesn't support Flatten layer,when I remove it ,I can import the model
     
    lijianfeng, Jan 27, 2021
    #1

  2. lijianfeng

    lijianfeng

    Joined:
    Sep 8, 2015
    Posts:
    54
    ok,reshape could do this.
     
    lijianfeng, Feb 4, 2021
    #2
    ROBYER1 likes this.

  3. alexandreribard_unity

    alexandreribard_unity

    Unity Technologies

    Joined:
    Sep 18, 2019
    Posts:
    53
    @lijianfeng I believe this is fixed in version 1.3.0
     
    alexandreribard_unity, Feb 19, 2021
    #3