search

liuliu / swift-diffusion

BSD 3-Clause "New" or "Revised" License
423 stars 33 forks source link

NHWC branch broken #28

Closed ghost closed 1 year ago

ghost commented 1 year ago 
tc @bazel-out/darwin_arm64-opt/bin/Diffusion.swiftmodule-0.params
src/Autoencoder.swift:12:91: error: type 'Convolution.Format?' has no member 'NCHW'
    hint: Hint(stride: [1, 1], border: Hint.Border(begin: [1, 1], end: [1, 1])), format: .NCHW)
                                                                                         ~^~~~
src/Autoencoder.swift:19:91: error: type 'Convolution.Format?' has no member 'NCHW'
    hint: Hint(stride: [1, 1], border: Hint.Border(begin: [1, 1], end: [1, 1])), format: .NCHW)
                                                                                         ~^~~~
src/Autoencoder.swift:23:97: error: type 'Convolution.Format?' has no member 'NCHW'
      groups: 1, filters: outChannels, filterSize: [1, 1], hint: Hint(stride: [1, 1]), format: .NCHW
                                                                                               ~^~~~
src/Autoencoder.swift:38:94: error: type 'Convolution.Format?' has no member 'NCHW'
    groups: 1, filters: inChannels, filterSize: [1, 1], hint: Hint(stride: [1, 1]), format: .NCHW)
                                                                                            ~^~~~
src/Autoencoder.swift:41:94: error: type 'Convolution.Format?' has no member 'NCHW'
    groups: 1, filters: inChannels, filterSize: [1, 1], hint: Hint(stride: [1, 1]), format: .NCHW)
liuliu commented 1 year ago

Should be OIHW. There is a rename in the latest s4nnc.

ghost commented 1 year ago

lmao what does O and I stand for?

liuliu commented 1 year ago

Output channel / Input channel.