Update model_finetune.py

model_finetune.py

@@ -19,10 +19,6 @@ from torch.optim.swa_utils import AveragedModel, SWALR
 from torch.utils.data import DataLoader, TensorDataset
 from model_regression_lsvq import Mlp, MAEAndRankLoss, preprocess_data, compute_correlation_metrics, logistic_func, plot_results
 
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-if device.type == "cuda":
-    torch.cuda.set_device(0)
-
 def create_results_dataframe(data_list, network_name, srcc_list, krcc_list, plcc_list, rmse_list, select_criteria_list):
     df_results = pd.DataFrame(columns=['DATASET', 'MODEL', 'SRCC', 'KRCC', 'PLCC', 'RMSE', 'SELECT_CRITERIC'])
     df_results['DATASET'] = data_list