microsoft
/
archai
зеркало из https://github.com/microsoft/archai.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность
2 633 коммитов 9 Ветки 12 Теги 57 MiB
Граф коммитов

166 Коммитов

Автор SHA1 Сообщение Дата
Wei-ge Chen ab87615d4f Continued clean up 2023-04-18 12:19:55 -07:00
Wei-ge Chen 1cd9f5ebdc search mode working 2023-04-17 16:26:59 -07:00
Wei-ge Chen db55484aaa trainer is working, search untested 2023-04-11 22:38:06 -07:00
Chris Lovett 28288861f1 add more info on docker image publish process for SNPE quantizer. 2023-04-07 16:58:58 -07:00
Chris Lovett 22a7430fd4 clarify instructions regarding use of windows for SNPE work. 2023-04-07 12:11:26 -07:00
Chris Lovett 66c9181cca Pass ArchaiStore as an input to RemoteAzureBenchmarkEvaluator. 
Make partition_key configurable in AzureStore.
 2023-04-06 22:14:10 -07:00
Chris Lovett de4590d3ba woops, fix bug. 2023-04-06 19:38:00 -07:00
Chris Lovett 9084e93b94 improvements to docker code and readmes 2023-04-06 19:33:06 -07:00
Chris Lovett d97f2e3695 factor out ensure_complete and record_error helper functions to simplify the code. 2023-04-06 18:17:58 -07:00
Chris Lovett 60358cbcb6 remove unnecessary handling of EdmType.INT64 since this is now handled by ArchaiStore. 2023-04-06 18:02:56 -07:00
Chris Lovett 933ed8ae60 remove branch from docker setup. 2023-04-06 17:08:10 -07:00
Chris Lovett b86871db4c fix all flake8 errors in face_segmentation 2023-04-06 15:41:30 -07:00
Chris Lovett 6b5c489fb2 remove launch.json 2023-04-06 14:14:00 -07:00
Chris Lovett 725d2f143b Merge branch 'task_segmentation' of github.com:microsoft/archai into task_segmentation 2023-04-05 19:28:56 -07:00
Chris Lovett a4524c28d8 fix bugs 2023-04-05 19:28:04 -07:00
Chris Lovett e2e7fe2592 fix flake8 errors 2023-04-05 18:50:09 -07:00
Chris Lovett c804f0a3f6 move RemoteAzureBenchmarkEvaluator to ArchaiStore 2023-04-05 15:41:13 -07:00
Chris Lovett 9c74e3a9cb fix bug 2023-04-05 15:34:53 -07:00
Chris Lovett 6dcbb29d07 fix cleanup when registry is already clean. 2023-04-05 13:33:00 -07:00
Chris Lovett 76641677ee add cleanup script. 2023-04-05 13:18:12 -07:00
Chris Lovett 10bd7024a8 Remove MODEL_STORAGE_CONNECTION_STRING from docker image and move it to the kubernetes deployment script. 2023-04-05 11:55:47 -07:00
Chris Lovett 38f7333e1c fix bugs in handing non-square images. 2023-04-05 11:24:02 -07:00
Chris Lovett dd7d6ef492 don't reset model_date 2023-04-04 21:46:24 -07:00
Chris Lovett 049ac46a4a fix run benchmark & readme setup instructions. 2023-04-04 21:34:49 -07:00
Chris Lovett b6ee9af9db Fix test_snpe so it works with new olive commands. 2023-04-04 21:24:52 -07:00
Chris Lovett 3a115cd048 fix reset preservation of benchmark_only flag. 2023-04-04 19:58:23 -07:00
Chris Lovett 38f35385e6 don't remove benchmark_only flag on reset. 2023-04-04 19:53:38 -07:00
Chris Lovett a2c7a89735 Remove ignored options from quant_options 
Rename quant.yaml to quantizer.yaml.
 2023-04-04 19:46:31 -07:00
Chris Lovett 0b30fa121b remove dead code, move CI pipeline to min python 3.8 which is needed by pytorch-lightning. 2023-04-04 19:30:15 -07:00
Chris Lovett 61a36ecf99 Merge branch 'task_segmentation' of github.com:microsoft/archai into task_segmentation 2023-04-04 18:18:35 -07:00
Chris Lovett d220dac9b2 update help message 2023-04-04 18:18:09 -07:00
Chris Lovett b76d7855a8 fix bugs 2023-04-04 18:10:48 -07:00
Chris Lovett 4edf5a73df port to public version of Olive. 2023-04-04 17:17:46 -07:00
Chris Lovett 5aa824c465 Merge branch 'clovett/snpe' into task_segmentation 2023-04-04 10:49:38 -07:00
Chris Lovett dc6677d61d fix bugs 2023-04-04 10:49:35 -07:00
Chris Lovett fca71be4f0 fix reference to usage table 2023-04-04 10:24:34 -07:00
Chris Lovett c9ce85792a fix bugs 2023-04-03 22:11:24 -07:00
Chris Lovett 280c94946f add snpe runner 2023-04-03 19:03:01 -07:00
Chris Lovett 7be38ec954 word wrap at 120 for easier diff reading. 
fix default search_config
Fix default val_check_interval
 2023-04-03 17:30:15 -07:00
Chris Lovett 2553298145 remove dataset_provider from search algo constructor. 2023-04-03 17:07:02 -07:00
Chris Lovett 496a27609c Merge branch 'main' into task_segmentation 2023-04-03 16:58:56 -07:00
Chris Lovett b689fd4399 Add TorchNumParameters constraint. 2023-04-03 16:51:24 -07:00
piero2c 39542d98a9 docs(task/seg): add final training results for SNP target 2023-04-03 07:40:29 -07:00
Chris Lovett 190b43a8d8 add tool to convert checkpoints to onnx. 2023-03-31 23:50:45 -07:00
Chris Lovett b80150587d move to latest pytorch-lightning>=2.0.0 2023-03-31 19:44:44 -07:00
Chris Lovett abfb0a528d export the onnx model in train.py 2023-03-31 18:28:04 -07:00
Chris Lovett c0d11c0931 add a python test script. 2023-03-31 17:21:06 -07:00
piero2c 3dcd2993cb feat(task/seg): set verbose to true in snp_search cfg 2023-03-31 11:22:56 -07:00
piero2c 94e2bb9fb2 feat(task/seg): update snp search results 2023-03-31 11:22:34 -07:00
piero2c df190bfea0 fix(task/seg): training script changes for compat w/ newer version of lightning 2023-03-30 11:15:20 -07:00
piero2c f5170158b6 docs(task/seg): update README.md 2023-03-28 09:40:06 -07:00
piero2c 8f04b1eb48 fix(task/seg): change constraints to avoid generating models that are too large 2023-03-28 08:59:54 -07:00
piero2c 9c90da59f9 docs(task/seg): update README.md with final training results and better descriptions 2023-03-21 09:11:49 -07:00
piero2c 8d33b27aab fix(task/seg): search constraints 2023-03-21 09:08:57 -07:00
piero2c e758ee60e1 feat(task/seg): add SNP latency objective 2023-03-20 15:12:11 -07:00
piero2c 41345fca8e docs(task/seg): update readme 2023-03-16 12:37:47 -07:00
piero2c 0cb708018e fix(text_gen): sets save_pareto_weights to false 2023-03-07 12:09:47 -08:00
piero2c 0ab7a1fa2a refactor(search): renames optimize_memory_usage to clear_evaluated_models 2023-03-07 11:40:16 -08:00
Gustavo Rosa 36ef82261e chore(discrete_search): Adds option for clearing stored architectures from memory. 2023-03-07 10:03:34 -08:00
Gustavo Rosa 8537ea1f06 fix(tasks): Fixes typo on text_generation README. 2023-03-02 11:50:08 -03:00
Gustavo Rosa 94aca143f8 fix(tasks): Fixes a typo on generate_text.py script. 2023-03-02 11:44:14 -03:00
Gustavo Rosa 6408066473 chore(tasks): Updates models URLs and generation script. 2023-03-02 11:42:58 -03:00
Gustavo Rosa 54058f9d77 fix(tasks): Re-writes some sentences in Text-Generation readme. 2023-02-27 11:26:50 -03:00
Gustavo Rosa 40fd39156d chore(tasks): Adds zero-shot evaluation pareto frontier plot. 2023-02-27 11:19:53 -03:00
piero2c 59f0e9dfee feat(task/seg): small changes to default search params 2023-02-13 06:39:52 -08:00
piero2c fc87a9e4a2 feat(task/seg): minor changes to the default search space 2023-02-13 06:39:14 -08:00
Gustavo Rosa ccb013f08f fix(tasks): Fixes link to ONNX memory plot. 2023-02-13 10:10:13 -03:00
piero2c 17f57078f8 fix(task/seg): checkpoint callback 2023-02-07 12:57:28 -08:00
piero2c 71a2acb8fe fix(task/seg): remove unused batch-size param 2023-02-07 12:13:33 -08:00
piero2c d999af0be4 feat(task/seg): bananas config 2023-02-07 11:57:39 -08:00
piero2c 4a3197c832 docs(task/seg): update readme 2023-02-07 09:59:10 -08:00
piero2c 97bf575a5a fix(task/seg): adds pareto architectures 2023-02-07 09:52:17 -08:00
piero2c f1673199e1 fix(task/seg): change default number of epochs 2023-02-07 09:49:31 -08:00
piero2c 3af5b273e9 docs(task/seg): updates README.md 2023-02-07 09:49:20 -08:00
piero2c 747a74210c fix(task/seg): training script cli args 2023-02-07 09:37:56 -08:00
piero2c cda2faff70 feat(task/seg): partial_tr_epochs param 2023-02-07 09:20:53 -08:00
piero2c 524966399e docs(task/seg): add readme images 2023-02-03 14:31:19 -08:00
piero2c 4080fa2ef3 Merge branch 'main' into task_segmentation 2023-02-03 14:02:14 -08:00
piero2c cd436eda51 fix(task/seg): update train.py 2023-02-03 10:16:26 -08:00
piero2c 9f4fbff8e2 feat(task/seg): increase ch_expansion_factor default values 2023-02-02 13:56:07 -08:00
piero2c 87909334ff fix(task/seg): search config 2023-02-02 13:55:35 -08:00
piero2c 3b1be1d29f chore(task/seg): change default search algo to evolution pareto 2023-02-02 12:56:58 -08:00
piero2c 4ff829ae9b fix(task/seg): fix `PartialTrainingValIOU` metric 2023-02-02 12:56:32 -08:00
piero2c f507634aeb fix(task/seg): number of epochs 2023-02-02 11:39:09 -08:00
piero2c e70eb95239 fix(task/segmentation): onnx conversion opset 2023-02-02 11:38:58 -08:00
piero2c dafbda723f feat(task/seg): search aux files 2023-02-02 10:45:59 -08:00
piero2c 7370cd7513 feat(task/seg): NAS script 2023-02-02 10:44:56 -08:00
piero2c 6c658f7fe2 chore(task/seg): better formatting 2023-02-02 08:06:08 -08:00
piero2c 16a0542bd0 feat(task/seg): stacked hourglass 2023-02-02 08:05:31 -08:00
piero2c c7609f52b3 feat(task/seg): f1-score logging 2023-02-02 06:27:37 -08:00
piero2c 4beb6a0f74 fix(task/seg): metrics ignore labels 2023-02-02 06:27:13 -08:00
Gustavo Rosa 44233d6131 chore(tasks): Uses HfDiskDatasetProvider in text_generation/train.py to speed up the training. 2023-02-02 11:14:47 -03:00
Gustavo Rosa fa16e8313c Merge branch 'main' of github.com:microsoft/archai into task-text-generation 2023-02-02 11:13:51 -03:00
piero2c 871de53655 feat(task/seg): F1 score 2023-02-02 05:33:57 -08:00
piero2c 593c347696 feat(task/seg): basic training code 2023-02-02 05:11:05 -08:00
piero2c 9893935d03 chore(task/seg): __init__.py 2023-02-01 12:03:58 -08:00
piero2c 663f928b75 feat(tasks/seg): pytorch lightning trainer 2023-02-01 12:02:44 -08:00
piero2c 2252f16fec feat(tasks/seg): segmentation metrics 2023-02-01 12:01:55 -08:00
piero2c dc356bcd6a fix(tasks/segmentation): Hgnet forward pass 2023-02-01 11:28:09 -08:00
Gustavo Rosa bec6cba801 fix(onnx_loader): Uses default provider is providers are not supplied. 2023-02-01 15:26:35 -03:00