.. |
batch-norm.go
|
added more building block modules
|
2022-02-15 12:40:28 +11:00 |
conv-transpose.go
|
corrected pointer receiver conversion in vision sub-packages and examples
|
2020-11-01 11:59:08 +11:00 |
conv.go
|
fixed conv3D config
|
2021-08-03 17:56:25 +10:00 |
func.go
|
converted to pointer receiver at tensor APIs, tensor and nn sub-packages
|
2020-10-31 19:25:32 +11:00 |
init.go
|
updated APIs changes
|
2021-07-23 00:54:41 +10:00 |
jit.go
|
example/jit-train: added
|
2021-01-02 16:49:36 +11:00 |
layer-norm.go
|
more changes to pointer receiver at nn sub-package and clean-up return variable
|
2020-11-03 11:56:04 +11:00 |
linear.go
|
converted to pointer receiver at vision sub-package
|
2020-10-31 22:11:50 +11:00 |
loss.go
|
reworked on resnet, densenet, added BCELoss, CrossEntropyLoss, changed DataLoader.Reset()
|
2021-07-14 10:38:11 +10:00 |
optimizer_test.go
|
feat(nn/rnn_test): added unit tests for rnn
|
2020-06-25 13:29:02 +10:00 |
optimizer.go
|
reworked on resnet, densenet, added BCELoss, CrossEntropyLoss, changed DataLoader.Reset()
|
2021-07-14 10:38:11 +10:00 |
other.go
|
added more building block modules
|
2022-02-15 12:40:28 +11:00 |
rnn_test.go
|
nn/rnn: corrected interface type conversion
|
2020-10-31 23:30:04 +11:00 |
rnn.go
|
fixed nn/rnn NewLSTM() clashed weight names
|
2022-01-21 11:05:11 +11:00 |
scheduler_test.go
|
added OneCycleLR
|
2021-06-12 18:21:43 +10:00 |
scheduler.go
|
export nn/scheduler DefaultSchedulerOptions()
|
2021-07-25 18:29:11 +10:00 |
sequential.go
|
fixed nn.Seq Forward nil pointer if layers length = 1
|
2021-05-15 17:50:08 +10:00 |
sparse_test.go
|
converted to pointer receiver at tensor APIs, tensor and nn sub-packages
|
2020-10-31 19:25:32 +11:00 |
sparse.go
|
converted to pointer receiver at tensor APIs, tensor and nn sub-packages
|
2020-10-31 19:25:32 +11:00 |
varstore_test.go
|
updated APIs changes
|
2021-07-23 00:54:41 +10:00 |
varstore.go
|
added nn.Path.Paths() and nn.VarStore.Summary() methods
|
2021-11-13 22:23:38 +11:00 |