-
Notifications
You must be signed in to change notification settings - Fork 5
Open
Description
do_pytorch.py calls loss = model(data, target) (not out = model(data); loss = criterion(out, target) ). This is made possible using the stuff in helpers_pytorch.py which wrap the "raw" pytorch model, into something that does specific to the problem (i.e. applies the right loss).
This is not done for SSD300...
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels