mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
Fix weigh bias parsing and better error logging
This commit is contained in:
@@ -1,6 +1,8 @@
|
||||
#ifndef CUDANET_ACTIVATION_FUNCTIONS_H
|
||||
#define CUDANET_ACTIVATION_FUNCTIONS_H
|
||||
|
||||
#include <cuda_runtime.h>
|
||||
|
||||
namespace CUDANet::Kernels {
|
||||
|
||||
/**
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
#ifndef CUDANET_CONVOLUTION_H
|
||||
#define CUDANET_CONVOLUTION_H
|
||||
|
||||
#include <cuda_runtime.h>
|
||||
|
||||
namespace CUDANet::Kernels {
|
||||
|
||||
/**
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
#ifndef CUDANET_MATMUL_H
|
||||
#define CUDANET_MATMUL_H
|
||||
|
||||
#include <cuda_runtime.h>
|
||||
|
||||
namespace CUDANet::Kernels {
|
||||
|
||||
/**
|
||||
|
||||
@@ -4,6 +4,7 @@
|
||||
#include <vector>
|
||||
|
||||
#include "layer.cuh"
|
||||
#include "activation.cuh"
|
||||
|
||||
namespace CUDANet::Layers {
|
||||
|
||||
|
||||
Reference in New Issue
Block a user