once-for-all icon indicating copy to clipboard operation
once-for-all copied to clipboard

[ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment

Results 60 once-for-all issues
Sort by recently updated
recently updated
newest added

when running train_ofa_net.py torch 1.11.0 torchvision 0.12.0

Hi,I found that the `expand_ratio` passed in mobilnetV2 will keep the same number in all stages and cells, is this a bug? https://github.com/mit-han-lab/once-for-all/blob/80a2e26ef04426213ad836b056234914d33a3c36/ofa/imagenet_classification/networks/proxyless_nets.py#L176

Hello, I want to optimize the OFA subnet model but I meet the error: ``` Traceback (most recent call last): File "mobile_test.py", line 129, in data_loader,bn_list[0],net_config)) File "mobile_test.py", line 62,...

Thank you for your great job. I have a question about the calculation of importance. Here in Once for all, the importance is calculated by the input dimension. https://github.com/mit-han-lab/once-for-all/blob/cfa0722d57e3a2391eb36b8cf613dd17ff7a32ae/ofa/imagenet_classification/elastic_nn/modules/dynamic_layers.py#L263 But...

I have trained OFA network on the custom dataset. However, how to train my own acc predictor and efficiency predictor

Could you share how to calculate mean, std and count in mobile_trim.yaml? And what does it meaning? I want to change the width_mult to 0.5, but tje output_feature will be...

Hi, I noticed that here(link1) add base acc to train accuracy predictor and set the last linear layer bias as False. For the same batch data, they have the same...

`/opt/vitis_ai/conda/envs/vitis-ai-pytorch/lib/python3.6/site-packages/ofa/imagenet_classification/run_manager/run_manager.py in 17 from ofa.utils import get_net_info, cross_entropy_loss_with_soft_target, cross_entropy_with_label_smoothing 18 from ofa.utils import AverageMeter, accuracy, write_log, mix_images, mix_labels, init_models ---> 19 from ofa.utils import MyRandomResizedCrop 20 21 __all__ = ['RunManager']...

When I train with two cards on my machine with `horovodrun -np 2 -H localhost:2 python train_ofa_net.py`, The code works fine. But I found that the memory of both processes...

I am trying to obtain specialised deployment with MobileNetV3 architecture, but I obtain the following the error. Kindly suggest on what must be done. The error originates from this code:...