fix loss imports
This commit is contained in:
parent
23051041ce
commit
21cee225c2
|
|
@ -2,7 +2,7 @@ from asyncio import base_tasks
|
|||
import torch
|
||||
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()]
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue