adnet_train.m

function adnet_train()
% ADNET_TRAIN Train the ADNet 
%
% Sangdoo Yun, 2017.

addpath('train/');
addpath(genpath('utils/'));
init_settings;
run(matconvnet_path);

rng(1004);

% Training stage 1: SL
opts.vgg_m_path = vgg_m_path;
[net, train_videos] = adnet_train_SL(opts);
save('./models/net_sl.mat', 'net')

% Training stage 2: RL
net = adnet_train_RL(net, train_videos, opts);
save('./models/net.mat', 'net')