From 91decc353dacfa9b5794a47e3ab072cd2c45cd39 Mon Sep 17 00:00:00 2001
From: wojciechbatruszewicz <wojtek.baltruszewicz@gmail.com>
Date: Tue, 27 Jun 2023 15:19:55 +0200
Subject: [PATCH] Fix train

---
 .gitignore |   4 ++++
 model.pt   | Bin 0 -> 1939 bytes
 train.py   |   6 +++---
 3 files changed, 7 insertions(+), 3 deletions(-)
 create mode 100644 .gitignore
 create mode 100644 model.pt

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..b0ec1bd
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,4 @@
+gender_classification_v7.csv
+gender_classification_val.csv
+gender_classification_test.csv
+gender_classification_train.csv
\ No newline at end of file
diff --git a/model.pt b/model.pt
new file mode 100644
index 0000000000000000000000000000000000000000..c17f2d355607f4647f8bdd99c0bab933a4167e08
GIT binary patch
literal 1939
zcmb7Ee@t6d6n>@j2c@tLHgrShKyV;X+OiQM>wSTk;`Yes3X+0urL=EtrBH7BibgF$
zG8ki779|pcPBU>~fSSl~ThjJE{K2MVI++VY7Kk$@Mn&BIaEl2p-uK$!YX=%{a{FG-
zJKuNDIo~<gK#AiCLM|unTgk*mqS<YuU4=FaV^Os>x%B1hiTaq;5ofi!T`t<nINdE?
zMU}@!duZE}PAj8;M6MBva!d+LebA~=Jx@FB4hF_q8MnvkP?>#<)8$o|J#?eb>9Uy_
zy2b1EnA@}pD9*_-$uP=`JuP+`#)lyay~}N3s>4PFl$b==M-8PWF@9QGeNaZ#Fi^ga
zA}A##>xK!Gj1qN2#SsS=Ng@N4Mwn#cQsu&FbXvSH*@4e;#5=-V8o16Oqm&F>&vSIa
z4V0)0riAgS4BTjhn<D9G@1>I##h`@`gc%;>7(|q^8*awI;3&3)S<)H!kP$u{i9#EW
zVrvvf5qt!5h#0ty=jehNTojpM{G$xaGQw=1A3kPEz?qnvX~x0{1h?~9;>uww0dxKE
zalVFaw8x93fq9h3%NGjg4)gu6z`<Lv_=Jac`Jl=V3&SR^ng^{~sP;om*urb!v^Tq*
zHmLQ(BHrrrHmVGiOx{<SrMQcg;I?7SZc-9*K{pku_2uz=Fa4f6{Y2vBaRhp@!PW0h
zlR(}{UQ2j~eAd@a)*q6CidRkr?j$aSrYa<C=N>OR@#1#2w77vCi|a%qwS&leZU8(r
zQUg9osRYzuZ|LyIoUU$XJa|8p$JP$!p$eac{N?4FAk{DirdpT4LP9#YQlE}S_nrqA
zM+0EEp%;{-WRmt<?}4`CjiHbAle(c>XLY$tL+Go?lOS*4RkYt!656vv1~#AjE4W~v
z3rtM1=+kvcY}xf?bo9V(w(>_dq`CG1@Z9b}eJ4gx@1HM#pU6>k>_P|9*S!fwKa&QY
z>9>cfgO@@3#R4+eaUP6~Cz4djYk{3-Ea+0{EciJypd0!jnRKZ)khNQ<!08NU@aXL-
zaJYRIEcTUxs)jSbbYLFzv$J5gcAM_RR5iGom%l4x-(57@CT7nrHnGQNmJz4O^vsT5
z4V)VPp-8b%5skiFd*5`&4Erj7zsSl)C)r70yIzm()aygP9hpaJNhkPvp$SZ+jE8On
zx3FJ|-wh5Y*C1C~9LdGkIXCl$8aElY|E*)vgz;(h<teMgSJ$$Md~Wz9wCj`~T~hs_
zvt&p>-Hl?@BXOf|KED|<4lL{BWkVpSd<*RFKg8GPqt~at{om&ot$zNZWS-~z@Pc!Y
zn<Z|qol_D~7Rf0xmilttDoL$U=e0~l3SRNI?N7(yleul6MBHQLP8ZH+GWQ{BMc?qT
zSYanAqi}oi8UJ7l)v-vS#?i>{@%<t|YGRQ>b)u0SxVQqOHWn$=AR0+XqL4+g$Z0|S
zlA`jgSwTh=Gf+}#e#B*~xM0Ou8hN|-G~?mp))h>^EsW!?Q9{H25(M|zKuLIaDIp~|
H|FG}h>q-w`

literal 0
HcmV?d00001

diff --git a/train.py b/train.py
index c4dad87..ab2bbbd 100644
--- a/train.py
+++ b/train.py
@@ -11,11 +11,11 @@ import argparse
 class MyNeuralNetwork(nn.Module):
     def __init__(self, *args, **kwargs) -> None:
         super(MyNeuralNetwork, self).__init__(*args, **kwargs)
-        self.fc1 = nn.Linear(32, 7)
+        self.fc1 = nn.Linear(7, 12)
         self.relu = nn.ReLU()
-        self.fc1 = nn.Linear(12, 64)
+        self.fc1 = nn.Linear(7, 12)
         self.relu = nn.ReLU()
-        self.fc2 = nn.Linear(64, 1)
+        self.fc2 = nn.Linear(12, 1)
         self.sigmoid = nn.Sigmoid()
     
     def forward(self, x):