mirror of
https://github.com/prise6/smart-iss-posts
synced 2024-05-21 14:56:32 +02:00
23 lines
423 B
Python
23 lines
423 B
Python
import tensorflow as tf
|
|
import os
|
|
import sys
|
|
from dotenv import find_dotenv, load_dotenv
|
|
from data_loader import TFRecordsLoader
|
|
from trainer_model import BaseTrainer
|
|
from base_model import BaseModel
|
|
|
|
|
|
def main():
|
|
load_dotenv(find_dotenv())
|
|
|
|
sess = tf.Session()
|
|
|
|
model = BaseModel()
|
|
data_loader = TFRecordsLoader()
|
|
|
|
trainer = BaseTrainer(sess, model, data_loader)
|
|
trainer.train()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|