mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
0fb58d76a1
Summary: + Add ArgMin support to Caffe2 to PyTorch converter + Using hypothesis to parameterize different conditions for test Test Plan: buck test //caffe2/torch/fb/model_transform/c2_convert:c2_pt_converter_test Reviewed By: houseroad Differential Revision: D25016203 fbshipit-source-id: 94489fcf1ed3183ec96f9796a5b4fb348fbde5bc |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| algebra.py | ||
| arg_scope.py | ||
| array_helpers.py | ||
| control_ops.py | ||
| conv.py | ||
| db_input.py | ||
| dropout.py | ||
| elementwise_linear.py | ||
| fc.py | ||
| nonlinearity.py | ||
| normalization.py | ||
| pooling.py | ||
| quantization.py | ||
| tools.py | ||
| train.py | ||