AIWintermuteAI/aXeleRate

Threads Error

Closed this issue · 1 comments

I try to run axelerate in local then i have an issues with thread.
This Error Appears.

import sys
sys.path.append('aXeleRate')
from axelerate import setup_training, setup_inference

from axelerate.networks.common_utils.augment import visualize_classification_dataset

visualize_classification_dataset('/datasets', num_imgs=10, img_size=224, augment=True)

config = {
    "model" : {
        "type":                 "Classifier",
        "architecture":         "MobileNet7_5",
        "input_size":           224,
        "fully-connected":      [100,50],
        "labels":               [],
        "dropout" : 		0.2
    },
     "weights" : {
            "full":   				"",
            "backend":   		    "imagenet",
            "save_bottleneck":      False
        
    },
    "train" : {
        "actual_epoch":         10,
        "train_image_folder":   "datasets",
        "train_times":          4,
        "valid_image_folder":   "datasets",
        "valid_times":          4,
        "valid_metric":         "val_accuracy",
        "batch_size":           32,
        "learning_rate":        1e-4,
        "saved_folder":   		"classifier",
        "first_trainable_layer": "dense_1",
        "augumentation":				True
    },
    "converter" : {
        "type":   				[]
    }
}

from keras import backend as K 
K.clear_session()
model_path = setup_training(config_dict=config)

from axelerate.networks.common_utils.convert import Converter
converter = Converter('k210', 'MobileNet7_5', 'datasets')
converter.convert_model(model_path)

Environment:
anaconda
Python 3.6
TF 2.3.1

Hi there!
I see two issues:

  1. When run locally, the recommended way to do that is by executing python scripts, e.g.
    python3 axelerate/train.py --config [config-file]
    and so on
  2. aXeleRate is not tested on Windows. It is likely that the converter parts won't work properly. It is indicted in the README, that currently only

Ubuntu 18.04 - other Linux distros might work, but not tested. Mac/Win OS are not supported

If you'd like to test and debug it for Windows, I'll leave the issue open for now and put 'help wanted' tag. If not, I'll close it for now.