MCPcopy Create free account

hub / github.com/YU1ut/imet-6th-soltuion / functions

Functions49 in github.com/YU1ut/imet-6th-soltuion

↓ 3 callersFunction_reduce_loss
(loss)
main.py:383
↓ 3 callersFunctionload_model
(model: nn.Module, path: Path)
utils.py:28
↓ 3 callersFunctionmake_loader
(df: pd.DataFrame, image_transform)
main.py:64
↓ 3 callersFunctiontrain
(args, model: nn.Module, criterion, *, params, train_loader, valid_loader, init_optimizer, use_cuda,
main.py:180
↓ 3 callersFunctionwrite_event
(log, step: int, **data)
utils.py:64
↓ 2 callersFunction_make_mask
(argsorted, top_n: int)
main.py:375
↓ 2 callersFunction_smooth
(ys, indices)
utils.py:165
↓ 2 callersFunctionbinarize_prediction
Return matrix of 0/1 predictions, same shape as probabilities.
main.py:362
↓ 2 callersFunctioncreate_net
(net_cls, pretrained: bool)
models.py:16
↓ 2 callersFunctionget_ids
(root: Path)
dataset.py:83
↓ 2 callersFunctionload_image
(item, root: Path)
dataset.py:77
↓ 2 callersFunctionmain
()
main.py:29
↓ 2 callersFunctionmean_df
(df: pd.DataFrame)
utils.py:24
↓ 2 callersFunctionpredict
(model, root: Path, df: pd.DataFrame, out_path: Path, batch_size: int, tta: int, workers: int, use
main.py:154
↓ 2 callersFunctionvalidation
( model: nn.Module, criterion, valid_loader, use_cuda, )
main.py:306
↓ 1 callersMethod__init__
(self, num_classes, pretrained=False, net_cls=M.resnet50, dropout=False)
models.py:28
↓ 1 callersMethod__init__
(self, root: Path, df: pd.DataFrame, image_transform: Callable, tta: int)
dataset.py:43
↓ 1 callersMethod_get_item
(self, i)
utils.py:55
↓ 1 callersFunctioncosine_rampdown
Cosine rampdown from https://arxiv.org/abs/1608.03983
main.py:356
↓ 1 callersFunctionget_score
(y_pred)
main.py:324
↓ 1 callersFunctionload_transform_image
( item, root: Path, image_transform: Callable, debug: bool = False)
dataset.py:59
↓ 1 callersFunctionmain
()
make_folds.py:32
↓ 1 callersFunctionmake_folds
(n_folds: int, seed: int)
make_folds.py:11
↓ 1 callersFunctiontrain_load_transform_image
( item, root: Path, image_transform: Callable, debug: bool = False)
dataset.py:68
Method__call__
(self, img)
transforms.py:28
Method__call__
(self, img)
transforms.py:62
Method__getitem__
(self, idx: int)
dataset.py:32
Method__getitem__
(self, idx)
dataset.py:53
Method__init__
(self, size, interpolation=Image.BILINEAR, min_aspect=4/5, max_aspect=5/4, m
transforms.py:18
Method__init__
(self, EPSILON = 0.5, sl = 0.02, sh = 0.4, r1 = 0.3, mean=[0.4914, 0.4822, 0.4465])
transforms.py:55
Method__init__
(self, num_classes, pretrained=False, net_cls=M.densenet121)
models.py:49
Method__init__
(self, root: Path, df: pd.DataFrame, image_transform: Callable, debug: bool = True)
dataset.py:21
Method__init__
(self, gamma=2)
main.py:388
Method__iter__
(self)
utils.py:36
Method__len__
(self)
dataset.py:29
Method__len__
(self)
dataset.py:50
Method_get_item
(self, i)
utils.py:60
Functionadjust_learning_rate
if epoch <= 1: lr = np.interp([epoch], [0, 1], [0, max_lr])[0] else:
main.py:342
Methodforward
(self, x)
models.py:12
Methodforward
(self, x)
models.py:44
Methodforward
(self, x)
models.py:60
Methodforward
(self, x)
main.py:76
Methodforward
(self, logit, target)
main.py:392
Methodfresh_params
(self)
models.py:41
Methodfresh_params
(self)
models.py:57
Functionget_classes
(item)
make_submission.py:33
Functiongmean_df
(df: pd.DataFrame)
utils.py:20
Functionmain
()
make_submission.py:10
Functionplot
Use in the notebook like this:: %matplotlib inline from imet.utils import plot plot('./runs/oc2', './runs/oc1', 'loss',
utils.py:72