Added ResNet and model selection

ai4iot / iot-segmentations

0 stars 0 forks source link

Closed msalexms closed 10 months ago

msalexms commented 10 months ago

New args implemented to select the models(for now efficientnet_b0 and resnet18) and arg name to store the results.