优化文件头

ant-research · Jul 7, 2020 · 9720b63 · 9720b63
1 parent ae57db3
commit 9720b63
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 6 deletions.
diff --git a/generate_distill_submission.py b/generate_distill_submission.py
@@ -1,11 +1,18 @@
+# @Author: yican, yelanlan
+# @Date: 2020-07-07 14:48:03
+# @Last Modified by:   yican
+# @Last Modified time: 2020-07-07 14:48:03
+# Standard libraries
 import pytorch_lightning as pl
 from pytorch_lightning.callbacks import EarlyStopping
 
+# Third party libraries
 import torch
 from scipy.special import softmax
 from torch.utils.data import DataLoader
 from tqdm import tqdm
 
+# User defined libraries
 from dataset import generate_transforms, PlantDataset
 from train import CoolSystem
 from utils import init_hparams, init_logger, seed_reproducer, load_data

diff --git a/generate_soft_labels.py b/generate_soft_labels.py
@@ -1,14 +1,21 @@
+# @Author: yican, yelanlan
+# @Date: 2020-07-07 14:47:29
+# @Last Modified by:   yican
+# @Last Modified time: 2020-07-07 14:47:29
+# Standard libraries
 import pandas as pd
 import pytorch_lightning as pl
 from pytorch_lightning.callbacks import EarlyStopping
 
+# Third party libraries
 import torch
 from dataset import generate_transforms
 from sklearn.model_selection import KFold
 from scipy.special import softmax
 from torch.utils.data import DataLoader
 from tqdm import tqdm
 
+# User defined libraries
 from train import CoolSystem
 from utils import init_hparams, init_logger, seed_reproducer, load_data
 from dataset import PlantDataset

diff --git a/loss_function.py b/loss_function.py
@@ -3,7 +3,6 @@
 # @Last Modified by:   NickYi1990
 # @Last Modified time: 2020-06-14 16:21:14
 # Third party libraries
-
 import torch.nn as nn
 import torch
 

diff --git a/train.py b/train.py
@@ -3,20 +3,19 @@
 # @Last Modified by:   yican.yc
 # @Last Modified time: 2020-06-16 20:36:19
 # Standard libraries
-# Third party libraries
-# User defined libraries
-
 import os
 import gc
 from time import time
 import pytorch_lightning as pl
 from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
 
+# Third party libraries
 import torch
 from dataset import generate_transforms, generate_dataloaders
 from sklearn.metrics import roc_auc_score
 from sklearn.model_selection import KFold
 
+# User defined libraries
 from models import se_resnext50_32x4d
 from utils import init_hparams, init_logger, seed_reproducer, load_data
 from loss_function import CrossEntropyLossOneHot

diff --git a/utils.py b/utils.py
@@ -2,8 +2,6 @@
 # @Date: 2020-06-14 16:19:48
 # @Last Modified by:   NickYi1990
 # @Last Modified time: 2020-06-30 10:11:22
-
-
 # Standard libraries
 import logging
 import os