Commit Graph

8 Commits

Author SHA1 Message Date
Artem Volkhin
ac7663b18c layer_model_instantiator: filter layers by tags
Summary: This diff allows to export a model partially, filtering layers by tags.

Reviewed By: kittipatv

Differential Revision: D4885610

fbshipit-source-id: 65394c5c9119d57a4d0703aa67ad8e79e4370e3b
2017-04-17 14:18:27 -07:00
Aaron Markham
58f7f2b441 doxygen python block added
Summary: Closes https://github.com/caffe2/caffe2/pull/226

Differential Revision: D4793550

Pulled By: JoelMarcey

fbshipit-source-id: cc33e58186304fa8dcac2ee9115dcc271d785b1e
2017-03-29 06:46:16 -07:00
Andrey Malevich
7cc92b1260 Add eval net for layer_model_helper
Summary:
This diff is adding eval nets to layer model helper. It should be useful for
the cases when train/eval nets need some extra input (usually some supervision)
for train/eval. For example various sampled layers, etc.

Differential Revision: D4769453

fbshipit-source-id: 7a8ec7024051eab73b8869ec21e20b5f10fd9acb
2017-03-29 04:03:40 -07:00
Kittipat Virochsiri
6163676ebe Skip optimizer when param doesn't have gradient and optimizer is not set
Summary: Currently, we cannot have layer constant because layer params are required to have gradient and optimizer. Global constants don't cut for this because it can only be added once; therefore, a layer that add any global constant can only be used once.

Differential Revision: D4773212

fbshipit-source-id: 5b60d31f3c1602afb04b61f6d30b8e3e06ed2de3
2017-03-24 22:18:34 -07:00
Andrey Malevich
84e742ded7 Migrate realtime training workflows to use new metrics.
Summary: This diff is getting rid of old metrics interface in realtime training.

Reviewed By: xianjiec

Differential Revision: D4649734

fbshipit-source-id: de4af85eb5476df9790ebd3915625bf8beee65af
2017-03-08 23:49:41 -08:00
Jiyan Yang
e75221e316 Add eval net to two tower workflow
Summary: The evaluation part of the two tower workflow is missing. This diff is to complete it. Part of the newly added functions can be used for other workflows, eg, feed. As the eval workflow in different workflows will be overlapped, a generic eval workflow will be added in a separate diff.

Reviewed By: kennyhorror

Differential Revision: D4646880

fbshipit-source-id: 4d6eb35df10f6f613533d442f2a04dc0332386f8
2017-03-07 21:03:00 -08:00
Dmytro Dzhulgakov
535e0e486b Add model graph to dper_example
Summary:
Just the first version displays forward part of the training net. I want to refactor local/distributed code to share graph initialization and then visualize all nets individually.

Graphs don't look pretty because of a lot of DotProducts, we need to refactor it.

Reviewed By: xianjiec

Differential Revision: D4514479

fbshipit-source-id: 156bb07c62118b15022c87f197b5e378a7ef3b9f
2017-02-07 13:03:54 -08:00
Yangqing Jia
238ceab825 fbsync. TODO: check if build files need update. 2016-11-15 00:00:46 -08:00