Fine tune training
This commit is contained in:
parent
9aac55f62d
commit
7cfc201229
3 changed files with 21677 additions and 14350 deletions
|
|
@ -13,7 +13,7 @@ import torch
|
|||
|
||||
MODELS = {
|
||||
# "v1": v1,
|
||||
"SimpleCNN": SimpleCNN,
|
||||
# "SimpleCNN": SimpleCNN,
|
||||
"Residual": Residual,
|
||||
# "NormalisedCNN": NormalisedCNN,
|
||||
# "SmartRes": SmartRes,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue