Fine tune training

This commit is contained in:
Andras Schmelczer 2024-06-06 08:21:34 +01:00
parent 9aac55f62d
commit 7cfc201229
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
3 changed files with 21677 additions and 14350 deletions

View file

@ -13,7 +13,7 @@ import torch
MODELS = {
# "v1": v1,
"SimpleCNN": SimpleCNN,
# "SimpleCNN": SimpleCNN,
"Residual": Residual,
# "NormalisedCNN": NormalisedCNN,
# "SmartRes": SmartRes,