attention.c File Reference

#include "darknet.h"
#include <sys/time.h>
#include <assert.h>

Include dependency graph for attention.c:

Go to the source code of this file.

Functions
void	extend_data_truth (data *d, int n, float val)
matrix	network_loss_data (network *net, data test)
void	train_attention (char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear)
void	validate_attention_single (char *datacfg, char *filename, char *weightfile)
void	validate_attention_multi (char *datacfg, char *filename, char *weightfile)
void	predict_attention (char *datacfg, char *cfgfile, char *weightfile, char *filename, int top)
void	run_attention (int argc, char **argv)

Function Documentation

extend_data_truth()

void extend_data_truth	(	data *	d,
		int	n,
		float	val
	)

Definition at line 6 of file attention.c.

network_loss_data()

matrix network_loss_data	(	network *	net,
		data	test
	)

Definition at line 18 of file attention.c.

predict_attention()

void predict_attention	(	char *	datacfg,
		char *	cfgfile,
		char *	weightfile,
		char *	filename,
		int	top
	)

Definition at line 382 of file attention.c.

run_attention()

void run_attention	(	int	argc,
		char **	argv
	)

Definition at line 434 of file attention.c.

train_attention()

void train_attention	(	char *	datacfg,
		char *	cfgfile,
		char *	weightfile,
		int *	gpus,
		int	ngpus,
		int	clear
	)

Definition at line 54 of file attention.c.

validate_attention_multi()

void validate_attention_multi	(	char *	datacfg,
		char *	filename,
		char *	weightfile
	)

Definition at line 321 of file attention.c.

validate_attention_single()

void validate_attention_single	(	char *	datacfg,
		char *	filename,
		char *	weightfile
	)

Definition at line 233 of file attention.c.