Update app.py

app.py

@@ -124,7 +124,7 @@ def lossFN(outs, targets):
     targets = targets.unsqueeze(1).expand_as(outs)
     return TNN.BCEWithLogitsLoss()(outs, targets)
 
-opt = TRNSFM.AdamW(params=mod.parameters(), lr=LEARNING_RATE)
+opt = torch.optim.Adam(mod.parameters(), lr=LEARNING_RATE)
 
 # Training and Finetuning
 def train(mod, training_loader):