Closed FinalCold closed 8 months ago
Traceback (most recent call last): File "/opt/conda/lib/python3.10/site-packages/optuna/study/_optimize.py", line 200, in _run_trial value_or_values = func(trial) File "/data/ephemeral/home/level1-imageclassification-cv-10/utils/hpo.py", line 136, in objective metrics = train(model, epochs, criterion, optimizer, scheduler) File "/data/ephemeral/home/level1-imageclassification-cv-10/utils/hpo.py", line 41, in train loss = criterion(outputs, labels) File "/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(args, **kwargs) File "/data/ephemeral/home/level1-imageclassification-cv-10/utils/loss.py", line 47, in forward assert y_pred.ndim == 1 AssertionError
위와 같은 에러가 발생합니다. F1Loss를 수정해야할 것 같습니다.
Traceback (most recent call last): File "/opt/conda/lib/python3.10/site-packages/optuna/study/_optimize.py", line 200, in _run_trial value_or_values = func(trial) File "/data/ephemeral/home/level1-imageclassification-cv-10/utils/hpo.py", line 136, in objective metrics = train(model, epochs, criterion, optimizer, scheduler) File "/data/ephemeral/home/level1-imageclassification-cv-10/utils/hpo.py", line 41, in train loss = criterion(outputs, labels) File "/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "/opt/conda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(args, **kwargs) File "/data/ephemeral/home/level1-imageclassification-cv-10/utils/loss.py", line 47, in forward assert y_pred.ndim == 1 AssertionError
위와 같은 에러가 발생합니다. F1Loss를 수정해야할 것 같습니다.