add_model_resnet18 #31
@@ -1,7 +0,0 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import torch.nn as nn
|
||||
|
||||
|
||||
def setup_criterion():
|
||||
return nn.MSELoss()
|
||||
Reference in New Issue
Block a user