def EfficientNetB6Expt(directories, workers, aug, extra_tag="", **kwargs):
model_kwargs = {"classes": 1}
regular_experiment(f"{extra_tag}EfficientNetB6{'AUG' if aug else ''}",
EfficientNetB6, directories, 528, workers, model_kwargs,
aug, **kwargs)
def ResNet152Expt(directories, workers, aug, extra_tag="", **kwargs):
model_kwargs = {"in_channels": 3, "classes": 1}
regular_experiment(f"{extra_tag}ResNet152{'AUG' if aug else ''}",
ResNet152, directories, 244, workers, model_kwargs, aug,
**kwargs)
def BasicCNNExpt(directories, workers, aug, extra_tag="", **kwargs):
model_kwargs = {}
regular_experiment(f"{extra_tag}BasicCNN{'AUG' if aug else ''}", BasicCNN,
directories, 512, workers, model_kwargs, aug, **kwargs)
def VGG16Expt(directories, workers, aug, extra_tag="", **kwargs):
model_kwargs = {"in_channels": 3, "output_size": 1}
regular_experiment(f"{extra_tag}VGG16{'AUG' if aug else ''}", VGG16,
directories, 244, workers, model_kwargs, aug, **kwargs)