https://github.com/samet-akcay/ganomaly
Revision 36e965af0c5db61bd8b5b1cc9d91f23754b02ee2 authored by akczay on 16 May 2018, 13:41:25 UTC, committed by akczay on 16 May 2018, 13:41:25 UTC
1 parent b85847b
Tip revision: 36e965af0c5db61bd8b5b1cc9d91f23754b02ee2 authored by akczay on 16 May 2018, 13:41:25 UTC
Removed test.py
Removed test.py
Tip revision: 36e965a
train.py
"""
TRAIN GANOMALY
. Example: Run the following command from the terminal.
run train.py \
--model ganomaly \
--dataset UCSD_Anomaly_Dataset/UCSDped1 \
--batchsize 32 \
--isize 256 \
--nz 512 \
--ngf 64 \
--ndf 64
"""
##
# LIBRARIES
from __future__ import print_function
from options import Options
from lib.data import load_data
from lib.model import Ganomaly
# ##
# def main():
# """ Training
# """
# ##
# # ARGUMENTS
# opt = Options().parse()
# ##
# # LOAD DATA
# dataloader = load_data(opt)
# ##
# # LOAD MODEL
# model = Ganomaly(opt, dataloader)
# ##
# # TRAIN MODEL
# model.train()
# if __name__ == '__main__':
# main()
##
# ARGUMENTS
opt = Options().parse()
##
# LOAD DATA
dataloader = load_data(opt)
##
# LOAD MODEL
model = Ganomaly(opt, dataloader)
##
# TRAIN MODEL
model.train()
Computing file changes ...