-
Notifications
You must be signed in to change notification settings - Fork 0
/
fei-nn.cabal
136 lines (129 loc) · 4.72 KB
/
fei-nn.cabal
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
cabal-version: 2.2
name: fei-nn
version: 2.0.0
synopsis: Train a neural network with MXNet in Haskell.
description: High level APIs to rain a neural network with MXNet in Haskell.
homepage: http://github.com/pierric/fei-nn
license: BSD-3-Clause
license-file: LICENSE
author: Jiasen Wu
maintainer: [email protected]
copyright: Copyright: (c) 2020 Jiasen Wu
category: Machine Learning, AI
build-type: Simple
Flag mxnet_geq_10600 {
Description: MXNet >= 1.6.0
Default: False
}
Flag mxnet_geq_10700 {
Description: MXNet >= 1.7.0
Default: False
}
Flag mxnet_geq_10800 {
Description: MXNet >= 1.8.0
Default: False
}
Flag neptune {
Description: Integrate Neptune
Default: False
}
Flag use_repa {
Description: Use Repa
Default: False
}
Library
if flag(use_repa) {
exposed-modules: MXNet.NN.Utils.Repa
build-depends: repa
}
exposed-modules: MXNet.NN
MXNet.NN.Types
MXNet.NN.Session
MXNet.NN.Utils
MXNet.NN.Utils.GraphViz
MXNet.NN.Utils.Render
MXNet.NN.Layer
MXNet.NN.Optimizer
MXNet.NN.LrScheduler
MXNet.NN.EvalMetric
MXNet.NN.Initializer
MXNet.NN.Callback
MXNet.NN.TaggedState
MXNet.NN.Module
MXNet.NN.DataIter.Class
MXNet.NN.DataIter.Vec
MXNet.NN.DataIter.Streaming
MXNet.NN.DataIter.Conduit
other-modules:
hs-source-dirs: src
ghc-options: -Wall
default-language: Haskell2010
default-extensions: GADTs,
TypeFamilies,
TypeOperators,
OverloadedLabels,
OverloadedStrings,
FlexibleContexts,
FlexibleInstances,
LambdaCase,
MultiWayIf,
DoAndIfThenElse,
TypeApplications,
DataKinds,
RecordWildCards,
ExplicitForAll,
ExistentialQuantification,
NoImplicitPrelude
if impl(ghc >= 8.6)
default-extensions: NoMonadFailDesugaring
build-depends: base >= 4.7 && < 5.0
, resourcet >= 1.1.8
, lens >= 4.12
, transformers-base >= 0.4.4
, aeson >= 1.2
, containers >= 0.5
, template-haskell >= 2.12
, graphviz
, time < 2.0
, type-combinators
, streaming >= 0.1.4.5
, conduit >= 1.2 && < 1.4
, conduit-combinators >= 1.1.2 && < 1.4
, constraints
, stm-conduit
, formatting
, wl-pprint-text
, Rasterific
, JuicyPixels
, FontyFruity
, rio
, uuid
, type-sets
, fei-base >= 2.0.0
if flag(mxnet_geq_10600) {
cpp-options: -DMXNET_VERSION=10600
}
if flag(mxnet_geq_10700) {
cpp-options: -DMXNET_VERSION=10700
}
if flag(mxnet_geq_10800) {
cpp-options: -DMXNET_VERSION=10800
}
if flag(neptune) {
cpp-options: -DNEPTUNE
build-depends: neptune-backend < 1
}
test-suite tests
type: exitcode-stdio-1.0
hs-source-dirs: test
main-is: TestMain.hs
other-modules: TestModule
build-depends: rio
, base
, lens
, tasty
, tasty-hunit
, tasty-discover
, fei-base >= 2.0.0
, fei-nn >= 2.0.0
default-language: Haskell2010