pytorchlib/ToDo at master · MarioProjects/pytorchlib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

 --------------------------
 #### IMPORTANCIA ALTA ####
 --------------------------

     -> Comentar todo estilo PEP8 (mirar proyecto Felix?)
     -> Hacer los examples como estan expuestos en el README
     -> Ejemplo de GAN en examples
     -> Ejemplo basico Unet y segmentación (mirar Kaggle y 1º solucion)

 ---------------------------
 #### IMPORTANCIA MEDIA ####
 ---------------------------

    -> Hacer genericas las redes (poder pasar listas de configuración):
        x BasicNets Convolucional
        x DenseNet
    -> Hacer posible que pasemos una lista de configuracion de la red para inicializarla (VGG, ResNet, DenseNet, BasicNets)
    -> Meter en los trainers la posibilidad de utilizar mixup

 --------------------------
 #### IMPORTANCIA BAJA ####
 --------------------------

    -> Comprobar cuda available para usar cuda y torch.cuda.floatensor y demas ->
            device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") DEPUES model.to(device)


 -----------------------
 #### RECORDATORIOS ####
 -----------------------