finetuning

fine_tune.py

@@ -203,9 +203,7 @@ for epoch in range(epochs):
     print(f'Train loss -- {train_loss} -- accuracy {train_acc} -- f1 {train_f1}')
 
     # save model
-    # !!!!!!!!!!!!!!!!
-    model_name = args.model.split('/')[-1] if args.model.split('/')[-1] != '' else args.model.split('/')[-2]
-    
+    model_name = args.model.split('/')[-1] if args.model.split('/')[-1] != '' else args.model.split('/')[-2]  
     torch.save(model.state_dict(),f'saved_models/{model_name}_epoch_{epochs}.bin')
 
     val_acc, val_loss, val_f1, report = eval_model(