Wasi Ahmad
Wasi Ahmad
¿Qué hace .view() en PyTorch?
¿ Qué le hace .view()a un tensor x? ¿Qué significan los valores negativos? x = x.view(-1, 16 * 5 * 5)
¿Cómo imprimo el resumen del modelo en PyTorch?
¿Cómo imprimo el resumen de un modelo en PyTorch como lo model.summary()hace en Keras? Model Summary: ____________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ==================================================================================================== input_1 (InputLayer) (None, 1,
¿Cómo guardo un modelo entrenado en PyTorch?
¿Cómo guardo un modelo entrenado en PyTorch? He leído que: torch.save()/ torch.load()es para guardar/cargar un objeto serializable. model.state_dict()/ model.load_state_dict()es para guardar/cargar el estado del modelo.