fix loss imports
This commit is contained in:
parent
23051041ce
commit
21cee225c2
|
|
@ -2,7 +2,7 @@ from asyncio import base_tasks
|
||||||
import torch
|
import torch
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from enhancer.utils.loss import mean_absolute_error, mean_squared_error
|
from enhancer.loss import mean_absolute_error, mean_squared_error
|
||||||
|
|
||||||
loss_functions = [mean_absolute_error(), mean_squared_error()]
|
loss_functions = [mean_absolute_error(), mean_squared_error()]
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue