Threads Error
SomprasitK opened this issue · 1 comments
SomprasitK commented
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
AIWintermuteAI commented
Hi there!
I see two issues:
- 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 - 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.