Commit f734e992 authored by Johan Gudmundsson's avatar Johan Gudmundsson
Browse files

update borcify nn for new droput module

parent 8b736e5d

src/borch/metrics/__init__.py

+6 −1
Original line number Diff line number Diff line
@@ -5,5 +5,10 @@ Metrics 
Module implementing metrics for torch tensors



"""

from borch.metrics.rv_metrics import all_metrics, mean_squared_error, accuracy

from borch.metrics.rv_metrics import (

    all_metrics,

    mean_squared_error,

    accuracy,

    module_metrics,

)

from borch.metrics import metrics

src/borch/nn/torch_proxies.py

+2 −0
Original line number Diff line number Diff line
@@ -99,6 +99,7 @@ _NO_WEIGHTS_MODULE_NAMES = [ 
    "CosineEmbeddingLoss",

    "CrossEntropyLoss",

    "Dropout",

    "Dropout1d",

    "Dropout2d",

    "Dropout3d",

    "ELU",
@@ -378,6 +379,7 @@ __all__ = [ 


_NEWER_MODULES = {

    "HuberLoss",

    "Dropout1d",

    "LazyBatchNorm1d",

    "LazyBatchNorm2d",

    "LazyBatchNorm3d",