From c2019ed7248b8b0430bc33e4c6d5366cba9d4c15 Mon Sep 17 00:00:00 2001
From: Dipam Chakraborty <dipam@aicrowd.com>
Date: Fri, 1 Apr 2022 21:42:37 +0530
Subject: [PATCH] No LR Scheduler + 25 Epochs + Unfreeze all layers +
Efficientnetb0
---
evaluator/model.py | 2 +-
local_evaluation.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/evaluator/model.py b/evaluator/model.py
index 483b332..74bf5ed 100644
--- a/evaluator/model.py
+++ b/evaluator/model.py
@@ -29,7 +29,7 @@ class ZEWDPCModel(torch.nn.Module):
def init_network(self):
# Setup Base Model - EfficientNet_b4
- self.base_model = torchvision.models.efficientnet_b4(
+ self.base_model = torchvision.models.efficientnet_b0(
pretrained=self.use_pretrained,
)
diff --git a/local_evaluation.py b/local_evaluation.py
index 4999c37..e3d85e1 100644
--- a/local_evaluation.py
+++ b/local_evaluation.py
@@ -131,7 +131,7 @@ else:
trainer = ZEWDPCTrainer(num_classes=6, use_pretrained=True)
trainer.train(
- aggregated_dataset, num_epochs=10, validation_percentage=0.1, batch_size=32
+ aggregated_dataset, num_epochs=25, validation_percentage=0.1, batch_size=64
)
y_pred = trainer.predict(val_dataset)
--
GitLab