- 14 May, 2019 3 commits
-
-
derekjchow authored
-
André Araujo authored
* Initial feature aggregation code for Detect-to-Retrieve paper. PiperOrigin-RevId: 246043144 * Add support for ASMK/ASMK*/R-ASMK/R-ASMK*. PiperOrigin-RevId: 247337028 * Add DatumProto uint32 field, and limit datum_io to uint32 and float32/float64 types. Also, introduce DatumPairProto, to be used for ASMK variants. Functions to read/write in this new format are added and tested. PiperOrigin-RevId: 247515205 * Add batching option to feature aggregation extraction. PiperOrigin-RevId: 247614627 * Script to perform local feature aggregation, with associated configs. Also small edits to the aggregation extractor, for better handling of input features / avoiding OOM. PiperOrigin-RevId: 248150750 * Tests to check that aggregation using regions with no local features works. PiperOrigin-RevId: 248153275 * Include new library/proto for aggregation * Merged commit includes the following changes: PiperOrigin-RevId: 248176511 * Merged commit includes the following changes: 248194572 by Andre Araujo: Change tf.tensor_scatter_nd_add --> tf.compat.v1.tensor_scatter_add to make it compatible with TF 1.X. -- PiperOrigin-RevId: 248194572 -
achowdhery authored
Create ssd_mobilenet_v2_fpnlite_quantized_shared_box_predictor_256x256_depthmultiplier_75_coco14_sync.config (#6775) Add config for FPNLite quantized model
-
- 11 May, 2019 2 commits
-
-
Toby Boyd authored
- Test passes locally python3 and test is already skipped for python2. -
Toby Boyd authored
* Add FP16 and benchmarks. * add missing run and report. * Add loss_scale as option not included with dtype. * move loss_scale validation under dtype conditional. * add loss_scale to flags tested.
-
- 10 May, 2019 5 commits
-
-
Haoyu Zhang authored
* Fix trivial model to work properly with fp16 * Add comment on manual casting
-
Haoyu Zhang authored
Previously we had one dense layer in trivial model. The weight was [224*224*3, num_classes]. Using two dense layers, the weights are [224*224*3, 1] and [1, num_classes].
-
Haoyu Zhang authored
* Do not report metrics in performance benchmarks * Rename flag
-
Haoyu Zhang authored
-
Haoyu Zhang authored
* Modified tweaked tests to use tensor learning rate
-
- 09 May, 2019 2 commits
-
-
Toby Boyd authored
* Add first benchmark and return stats. * Remove print statements update training steps. * Revert print T: in print statement. * Remove print(stats) * add 2 gpu accuracy test for base. * Fixed total_batch_size when using gpu + gFile deprecations. * 8 GPU test name fix * Add 4 and 8 GPU tests. * typo fixes. * Clean up test names and methods. * bleu uncased. docstring format fix.
-
Haoyu Zhang authored
* Add learning rate tensor. This makes training slower * Improve LearningRateSchedule with better efficiency * Fix lint error * Replace constant definition with existing one
-
- 08 May, 2019 2 commits
-
-
Toby Boyd authored
-
Menglong Zhu authored
246873701 by menglong: Missing __init__.py under meta_architectures/ -- 246857392 by menglong: Standardize proto namespace: lstm_object_detection.protos -- 246625127 by menglong: Internal changes. -- 246596481 by menglong: Add License -- 246580605 by menglong: Internal changes -- 246344626 by menglong: Open source interleaved mobilenet v2 model. -- 244893883 by menglong: Introduce multi_input_decoder for interleaved model. -- 244461016 by menglong: Add pre-bottleneck operation to lstm cells to support interleaved model. -- 244052176 by menglong: Update README -- 244020495 by menglong: Add test to rnn_decoder. -- 243704250 by menglong: Duplicate assignment. -- 243091836 by menglong: Move LSTMSSD meta arch into separate folder -- 242900337 by menglong: Modified mobilenet definition for LSTM-SSD -- 242773195 by menglong: Release GroupedConvLSTMCell implementation: https://arxiv.org/abs/1903.10172 -- 242574736 by menglong: Introduce module for quantizated training. -- 242544306 by menglong: lstm_ssd_meta_arch updates, added test rename: - LSTMMetaArch to LSTMSSDMetaArch - LSTMFeatureExtractor to LSTMSSDFeatureExtractor -- 241986236 by menglong: Move lstm quantization utils to 3rd party. -- 225922488 by yinxiao: Training pipeline fixes. -- 224839137 by yinxiao: Issue fix for lstm object detecion sample config. -- 224246947 by menglong: Fix logging module import -- PiperOrigin-RevId: 246873701
-
- 07 May, 2019 2 commits
-
-
Haoyu Zhang authored
-
Toby Boyd authored
-
- 06 May, 2019 2 commits
-
-
Yukun Zhu authored
* deeplab quantize * Fix bug in train.py * Create quantize.md
-
Haoyu Zhang authored
-
- 04 May, 2019 1 commit
-
-
Haoyu Zhang authored
* Enable CuDNN BatchNorm spatial persistent by default; Remove 2nd zero padding layer * Apply scale=False and fused=True consistently to BatchNorm layers * Undo remove padding layer * Replace zero padding with padding attribute in max pooling for better performance * Resolve comments * Revert "Replace zero padding with padding attribute in max pooling for better performance" This reverts commit ad49db057c800ecac008eec1057005bd2c08ac73.
-
- 03 May, 2019 1 commit
-
-
Reed authored
-
- 02 May, 2019 2 commits
-
-
Krishnan Srinivasan authored
* updated mujoco api calls to data, and tensorflow call to CriticalSection (moved from contrib.framework) * changed efficient-hrl envs so self.physics points to correct PyMj object based on mujoco_py version * corrected mujoco version number checkm in AntEnv.physics and PointEnv.physics properties * fix AntEnv.set_xy by reverting to using self.physics
-
Reed authored
-
- 01 May, 2019 2 commits
-
-
Reed authored
This options allows the new tf.train.experimental.enable_mixed_precision_graph_rewrite() function to be used for fp16, instead of manual casts.
-
André Araujo authored
* Internal change. PiperOrigin-RevId: 185564155 * Fix small bug when reading DELF features when file is empty. A new test is added that catches this bug. PiperOrigin-RevId: 213839503 * Refactors DELF example code to expose a function to create a DELF feature extractor. PiperOrigin-RevId: 241492615 * Merged commit includes the following changes: 244073180 by Andre Araujo: Internal change -- 243646498 by Andre Araujo: Detect and save bounding boxes for a list of images. A new proto is added, along with auxiliary read/write functions, and tests. -- PiperOrigin-RevId: 244073180 * Merged commit includes the following changes: 244872693 by Andre Araujo: Small change. -- 244871213 by Andre Araujo: Small edit to DELF extractor function. -- 244790715 by Andre Araujo: internal automated change -- 244773146 by Andre Araujo: Add option to visualize extracted boxes, and change slightly model input. -- 244288992 by Andre Araujo: Script to cluster DELF features using K-means. -- 244275164 by Andre Araujo: Scripts to extract DELF features and boxes from Revisited Oxford/Paris datasets, used in Detect-to-Retrieve paper. -- PiperOrigin-RevId: 244872693 * Merged commit includes the following changes: PiperOrigin-RevId: 244876167 * Updates instructions to include new released models, instructions for DELF and detector extraction, etc. * Small fix * Merged commit includes the following changes: 244909336 by Andre Araujo: Update to DELF config example to use latest model. -- PiperOrigin-RevId: 244909336 * Revert "Merged commit includes the following changes:" This reverts commit f41be9f159330f62846d5d8b3d3d5d55cd1874f4. * Update delf_config_example model * Small edits to detection instructions * Specifying GLB location in README * Mention D2R code release in progress
-
- 30 Apr, 2019 1 commit
-
-
Toby Boyd authored
-
- 29 Apr, 2019 5 commits
-
-
Igor authored
Replace per_device with per_replica and PerDevice with PerReplica, because the PerDevice concept was renamed and doesn't exist anymore. (#6693) * Replace per_device with per_replica and PerDevice with PerReplica, because the PerReplica concept was renamed and doesn't exist anymore.
-
Toby Boyd authored
* Add accuracy check. * Avoid double flag init, move data_dir to real data. * Comment on lower accuracy target.
-
Shining Sun authored
* bug fix * bug fix
-
Igor authored
* Add benchmarks with the --cloning flag to Resnet and NFC. * Renamed cloning to clone_model_in_keras_dist_strat. Dropped a few tests that aren't essential. * Fixed up the formatting after re-naming the flag to a much longer name. Thanks, lint. * Fixed the lint error in nfc_common.py
-
Songyi Blair Han authored
-
- 26 Apr, 2019 4 commits
-
-
Toby Boyd authored
* Combined imagenet and cifar-10 benchmarks * Comments and epochs_between_evals. * Added tuned tests and cleaned up benchmark flags * Fix names. * Return results and add images/sec hook. * updated doc strings for return values. * 128 to 256 batch for FP16 test * added more doc strings to fix lint.
-
pkulzc authored
* Merged commit includes the following changes: 244869387 by Sergio Guadarrama: This CL adds script/code to generate Visual WakeWords Dataset annotation files and TF records starting from COCO dataset. -- 244866660 by Sergio Guadarrama: Add VisualWakeWords Dataset to Slim dataset_factory to train MobileNets on it. -- 244836000 by Sergio Guadarrama: n/a -- 244104396 by Sergio Guadarrama: Add an option whether to enable / disable image cropping in inception_preprocessing. -- 242040128 by Sergio Guadarrama: Internal change 241793677 by Sergio Guadarrama: Internal change 241073081 by Sergio Guadarrama: Internal change 240131189 by Sergio Guadarrama: Internal change PiperOrigin-RevId: 244869387 * Merged commit includes the following changes: 245431876 by Sergio Guadarrama: Internal cleanup -- PiperOrigin-RevId: 245431876 * Merged commit includes the following changes: 245454983 by Sergio Guadarrama: Internal Cleanup -- PiperOrigin-RevId: 245454983 -
Ayush Dubey authored
* Add num_packs flag for MirroredStrategy's cross device ops. * fix parens * Fix lint errors and make all_reduce_alg more robust. * Set default num_packs to 1
-
Gaurav Jain authored
tf.test.is_gpu_available() should not be called in flags since this is called before app.main() and the runtime has not yet been initialized.
-
- 25 Apr, 2019 2 commits
-
-
Ayush Dubey authored
* Remove contrib AllReduceCrossDeviceOps and update all_reduce_alg options with MirroredStrategy. * cleanup
-
Haoyu Zhang authored
Reason: test failures because contrib is not available in V2 This reverts commit 325dd761.
-
- 24 Apr, 2019 4 commits
-
-
Toby Boyd authored
-
Shining Sun authored
* Added none check for output_dir * Change double quote to single
-
André Araujo authored
* Internal change. PiperOrigin-RevId: 185564155 * Fix small bug when reading DELF features when file is empty. A new test is added that catches this bug. PiperOrigin-RevId: 213839503 * Refactors DELF example code to expose a function to create a DELF feature extractor. PiperOrigin-RevId: 241492615 * Merged commit includes the following changes: 244073180 by Andre Araujo: Internal change -- 243646498 by Andre Araujo: Detect and save bounding boxes for a list of images. A new proto is added, along with auxiliary read/write functions, and tests. -- PiperOrigin-RevId: 244073180 * Merged commit includes the following changes: 244872693 by Andre Araujo: Small change. -- 244871213 by Andre Araujo: Small edit to DELF extractor function. -- 244790715 by Andre Araujo: internal automated change -- 244773146 by Andre Araujo: Add option to visualize extracted boxes, and change slightly model input. -- 244288992 by Andre Araujo: Script to cluster DELF features using K-means. -- 244275164 by Andre Araujo: Scripts to extract DELF features and boxes from Revisited Oxford/Paris datasets, used in Detect-to-Retrieve paper. -- PiperOrigin-RevId: 244872693 * Merged commit includes the following changes: PiperOrigin-RevId: 244876167 * Updates instructions to include new released models, instructions for DELF and detector extraction, etc. * Small fix * Merged commit includes the following changes: 244909336 by Andre Araujo: Update to DELF config example to use latest model. -- PiperOrigin-RevId: 244909336 * Revert "Merged commit includes the following changes:" This reverts commit f41be9f159330f62846d5d8b3d3d5d55cd1874f4. * Update delf_config_example model * Small edits to detection instructions * Specifying GLB location in README -
Haoyu Zhang authored
-